diff --git a/5. HyperParameter Tuning and Transfer Learning/5.1 HyperParameter Tuning.ipynb b/5. HyperParameter Tuning and Transfer Learning/5.1 HyperParameter Tuning.ipynb
index c5912f8..f6a4b37 100644
--- a/5. HyperParameter Tuning and Transfer Learning/5.1 HyperParameter Tuning.ipynb	
+++ b/5. HyperParameter Tuning and Transfer Learning/5.1 HyperParameter Tuning.ipynb	
@@ -102,7 +102,7 @@
     "# load training data and do basic data normalization\n",
     "(X_train, y_train), (X_test, y_test) = mnist.load_data()\n",
     "\n",
-    "if K.image_dim_ordering() == 'th':\n",
+    "if  K.image_data_format() == 'th':\n",
     "    X_train = X_train.reshape(X_train.shape[0], 1, img_rows, img_cols)\n",
     "    X_test = X_test.reshape(X_test.shape[0], 1, img_rows, img_cols)\n",
     "    input_shape = (1, img_rows, img_cols)\n",