diff --git a/Keras Adversarial Autoencoder MNIST.ipynb b/Keras Adversarial Autoencoder MNIST.ipynb
index 3abbdb3..c9d8c8f 100644
--- a/Keras Adversarial Autoencoder MNIST.ipynb	
+++ b/Keras Adversarial Autoencoder MNIST.ipynb	
@@ -261,7 +261,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def sample_prior(latent_dim, batch_size):\n",
+    "def sample_prior(batch_size, latent_dim):\n",
     "    return np.random.normal(size=(batch_size, latent_dim))"
    ]
   },
@@ -862,7 +862,7 @@
     "    latent_fake = encoder.predict(imgs)\n",
     "    \n",
     "    # Here we generate the \"TRUE\" samples\n",
-    "    latent_real = sample_prior(latent_dim, batch_size)\n",
+    "    latent_real = sample_prior(batch_size, latent_dim)\n",
     "                      \n",
     "    # Train the discriminator\n",
     "    d_loss_real = discriminator.train_on_batch(latent_real, valid)\n",
@@ -921,4 +921,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file