about summary refs log tree commit diff stats
diff options
context:
space:
mode:
-rw-r--r--main.lua6
1 files changed, 5 insertions, 1 deletions
diff --git a/main.lua b/main.lua
index 11835bc..658d94d 100644
--- a/main.lua
+++ b/main.lua
@@ -93,7 +93,11 @@ function love.draw()
       if pmx > 16 and pmx < 16+drawingw and pmy > line.y and pmy < line.y+pixels(line.h) then
         love.graphics.setColor(0.75,0.75,0.75)
         love.graphics.rectangle('line', 16,line.y, drawingw,pixels(line.h))
-        icon[current_mode](16+drawingw-20, line.y+4)
+        if icon[current_mode] then
+          icon[current_mode](16+drawingw-20, line.y+4)
+        else
+          icon[previous_mode](16+drawingw-20, line.y+4)
+        end
 
         if love.mouse.isDown('1') and love.keyboard.isDown('h') then
           draw_help_with_mouse_pressed(line)