diff --git a/simple_regression.py b/simple_regression.py
index fa31f96..6097c23 100644
--- a/simple_regression.py
+++ b/simple_regression.py
@@ -48,7 +48,7 @@ history = model.fit(
 y_pred = model.predict(test_piece_counts)
 
 # Zapis predykcji do pliku
-results = pd.DataFrame({'test_set_piece_count': test_piece_counts.tolist(), 'predicted_price': [a[0].round(2) for a in y_pred.tolist()]})
+results = pd.DataFrame({'test_set_piece_count': test_piece_counts.tolist(), 'predicted_price': [round(a[0], 2) for a in y_pred.tolist()]})
 results.to_csv('lego_reg_results.csv', index=False, header=True)
 
 # Zapis modelu do pliku