X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fwidgets%2Frobot.py;h=09fd099f0a258d7d5136d590318512b4f7ca72f9;hb=83850d913f756713d0e82c768147a018e2dcc84d;hp=2eeb1d39cd2d663d514989a17a4e6557e9e7a80b;hpb=0cef06f581d48315a59a19ac06ca1d2be14cbb4c;p=naja.git

diff --git a/naja/widgets/robot.py b/naja/widgets/robot.py
index 2eeb1d3..09fd099 100644
--- a/naja/widgets/robot.py
+++ b/naja/widgets/robot.py
@@ -1,18 +1,16 @@
 """Widget to draw the player on the screen"""
 
-import pygame.locals as pgl
-
 from naja.constants import (PLAYER_SIZE, BIT_SIZE, TILE_SIZE, BITS)
-from naja.events import finish_event
 from naja.resources import resources
 from naja.resources.mutators import EIGHT_BIT
 from naja.widgets.base import Widget
 
 IMG_MAP = {
-        BITS.CYAN: 'board/robot_cyan.png',
-        BITS.YELLOW: 'board/robot_yellow.png',
-        BITS.MAGENTA: 'board/robot_magenta.png',
-        }
+    BITS.MSB: 'board/robot_msb.png',
+    BITS.RED: 'board/robot_red.png',
+    BITS.GREEN: 'board/robot_green.png',
+    BITS.BLUE: 'board/robot_blue.png',
+}
 
 
 class RobotWidget(Widget):