diff --git a/cv5/a_train.c b/cv5/a_train.c
index ccf3980..8c91f2b 100644
--- a/cv5/a_train.c
+++ b/cv5/a_train.c
@@ -52,21 +52,14 @@ struct car* clear_train(struct car* first, const char* target) {
         return 0;
     } 
 
-    if(first->next == NULL){
-      int result = strcmp(first->value, target);
-        if(result == 0){
-           free(first);
-           return 0;          
-        }
-        else{
-            return first;
-        }
-    }
-
     struct car* prev = first;
     while(prev->next != NULL){
         int x = strcmp(first->value, target);
         if(x == 0){
+            if(first->next == NULL){
+                free(first);
+                return NULL:
+            }  
             prev = first->next;
             free(first);
             return prev;