diff --git a/POET_Training.ipynb b/POET_Training.ipynb
index 6d6dd9d..e15872c 100644
--- a/POET_Training.ipynb
+++ b/POET_Training.ipynb
@@ -307,7 +307,7 @@
     "        df_design[key] = np.vectorize(func_dict[key])(df_design[key])\n",
     "        df_result[key] = np.vectorize(func_dict[key])(df_result[key])\n",
     "        \n",
-    "    return df_result, df_design\n",
+    "    return df_design, df_result\n",
     "\n",
     "# Get minimum and maximum values for each column\n",
     "def get_min_max(df_design, df_result):\n",