diff --git a/du4/a_train.c b/du4/a_train.c
index 37f66b5..a748883 100644
--- a/du4/a_train.c
+++ b/du4/a_train.c
@@ -54,6 +54,11 @@ struct car* clear_train(struct car* first, const char* target) {
         return first;
     }
     if(first->next->next == NULL && strcmp(first->value, target) == 0){
+       struct car* temp = first;
+       first = first->next;
+       first->next = NULL;
+       free(temp);
+       return NULL;
 
     }
     struct car* temp = first;
@@ -72,6 +77,6 @@ struct car* clear_train(struct car* first, const char* target) {
     
     }
     
-	return temp;
+	return first;
 }