diff --git a/cv5/a_train.c b/cv5/a_train.c
index a2152c2..d4b643e 100644
--- a/cv5/a_train.c
+++ b/cv5/a_train.c
@@ -73,13 +73,14 @@ struct car* clear_train(struct car* first, const char* target) {
                 free(this);
                 return first;
             }
-            else if(this->next->next == NULL){
+            else {
                 while(tmp->next != NULL){
                     prev = tmp;
                     tmp = tmp->next;
                 }
                 free(tmp);
-                prev->next = NULL;            
+                prev->next = NULL;  
+                return this;          
             }
         }
         this = this->next;