diff --git a/function.py b/function.py
index dff80b6..16476ff 100644
--- a/function.py
+++ b/function.py
@@ -123,7 +123,6 @@ def predict_logreg(test_X, weights):
 
     return predictions
 
-if __name__ == "__main__":
 if __name__ == "__main__":
     train_path = "./data/horse-colic.data"
     test_path = "./data/horse-colic.test"