diff --git a/pandasai/__init__.py b/pandasai/__init__.py
index 24b39b058..60781bdf0 100644
--- a/pandasai/__init__.py
+++ b/pandasai/__init__.py
@@ -172,9 +172,7 @@ def run_code(
         lines = code.strip().split("\n")
         last_line = lines[-1].strip()
         if last_line.startswith("print(") and last_line.endswith(")"):
-            # Last line is already printing
-            return eval(last_line[6:-1])
-        # Evaluate last line and return its value or the captured output
+            last_line = last_line[6:-1]
         try:
             result = eval(last_line)
             return result
diff --git a/tests/test_pandasai.py b/tests/test_pandasai.py
index d327159d3..88781e40f 100644
--- a/tests/test_pandasai.py
+++ b/tests/test_pandasai.py
@@ -164,4 +164,11 @@ def test_run_without_privacy_enforcement(self):
 to get the answer to the following question :
 How many countries are in the dataframe?"""
         self.pandasai.run(df, "How many countries are in the dataframe?")
-        assert self.pandasai._llm.last_prompt == expected_prompt
\ No newline at end of file
+        assert self.pandasai._llm.last_prompt == expected_prompt
+
+    def test_run_with_print_at_the_end(self):
+        code = """
+result = {"happiness": 0.5, "gdp": 0.8}
+print(result)"""
+        self.setup(code)
+        self.pandasai.run_code(code, pd.DataFrame())