diff --git a/paint_palette.py b/paint_palette.py
index e0371f4a6d601fcee3637b9be1c555ce7ed9cd87..7b9741097c14718768c9d7ba8e911011990937ae 100644
--- a/paint_palette.py
+++ b/paint_palette.py
@@ -67,7 +67,7 @@ def current_brush():
         brush = context.tool_settings.vertex_paint.brush
     elif context.area.type == 'VIEW_3D' and context.image_paint_object:
         brush = context.tool_settings.image_paint.brush
-    elif context.area.type == 'IMAGE_EDITOR' and  context.space_data.use_image_paint:
+    elif context.area.type == 'IMAGE_EDITOR' and  context.space_data.mode == 'PAINT':
         brush = context.tool_settings.image_paint.brush
     else :
         brush = None