diff --git a/lava-collector b/lava-collector
index 00c3690..4c9f2a2 100644
--- a/lava-collector
+++ b/lava-collector
@@ -29,7 +29,10 @@ local function findFirstLavaBucket()
 end
 
 local function returnToBase()
-    print(ForwardCounter)
+    print('returing to base')
+    for _ = 1, ForwardCounter do
+        turtle.back()
+    end
 end
 
 local function despositRefill()
@@ -70,6 +73,8 @@ while true do
         if compareItemName("minecraft:bucket") then
             if turtle.forward() then
                 ForwardCounter = ForwardCounter + 1
+            else
+                returnToBase()
             end
         end
     end