1 files changed, 9 insertions, 10 deletions
diff --git a/main.py b/main.py
index 52063f7..a0659f3 100755
--- a/main.py
+++ b/main.py
@@ -18,16 +18,15 @@ while running:
     for event in pygame.event.get():
         if event.type == pygame.QUIT:
             running = False
-
-    keys = pygame.key.get_pressed()
-    if keys[pygame.K_w] and view.game.snake.heading != Vec.up.neg():
-        view.nextControlDirection = Vec.up
-    elif keys[pygame.K_a] and view.game.snake.heading != Vec.left.neg():
-        view.nextControlDirection = Vec.left
-    elif keys[pygame.K_r] and view.game.snake.heading != Vec.down.neg():
-        view.nextControlDirection = Vec.down
-    elif keys[pygame.K_s] and view.game.snake.heading != Vec.right.neg():
-        view.nextControlDirection = Vec.right
+        elif event.type == pygame.KEYDOWN:
+            if event.key == pygame.K_w and view.game.snake.heading != Vec.up.neg():
+                view.nextControlDirection = Vec.up
+            elif event.key == pygame.K_a and view.game.snake.heading != Vec.left.neg():
+                view.nextControlDirection = Vec.left
+            elif event.key == pygame.K_r and view.game.snake.heading != Vec.down.neg():
+                view.nextControlDirection = Vec.down
+            elif event.key == pygame.K_s and view.game.snake.heading != Vec.right.neg():
+                view.nextControlDirection = Vec.right
 
     if view.isRunning():
         view.update(pygame.time.get_ticks())