diff --git a/problema_4.py b/problema_4.py
index c0d9e1e..50f293a 100644
--- a/problema_4.py
+++ b/problema_4.py
@@ -1,11 +1,13 @@
 altura = int(input("Ingresa la altura del pozo (metros): "))
-avanza = int(input('Ingresa cuanto avanza la rana de día (metros): '))
+avanza = int(input("Ingresa cuanto avanza la rana de día (metros): "))
 retrocede = int(input("Ingresa cuanto retrocede la rana de noche (metros): "))
 avance = avanza - retrocede
 avance_actual = 0
 dias = 0
 
 while altura > avance_actual:
+    if (avance_actual + avance) >= altura:
+        break
     avance_actual += avance
     dias += 1