diff --git a/du7/snake.c b/du7/snake.c
index 0d6441c..b0b1612 100644
--- a/du7/snake.c
+++ b/du7/snake.c
@@ -29,7 +29,11 @@ struct snake* add_snake(struct snake* snake,int x,int y){
 struct snake* remove_snake(struct snake* snake){
     struct snake* temp = snake;
 
-    if(snake == NULL || snake->next == NULL){
+    if(snake == NULL){
+        return snake;
+    }
+    else if(snake->next == NULL){
+        free(snake);
         return snake;
     }