From 46d03ac3ad70b18707abe2e1885bf98176545895 Mon Sep 17 00:00:00 2001
From: Maryna Kravtsova <mk757ye@omega.tuke.sk>
Date: Wed, 4 Nov 2020 23:20:00 +0100
Subject: [PATCH] clear

---
 cv5/a_train.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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;