diff --git a/assignment3/Assignment3.ipynb b/assignment3/Assignment3.ipynb
index 755c595..7459e8f 100644
--- a/assignment3/Assignment3.ipynb
+++ b/assignment3/Assignment3.ipynb
@@ -331,7 +331,7 @@
         "\n",
         "def make_dataset(N, theta=true_poly_theta, sigma=0.1):\n",
         "    \"\"\" Sample a dataset \"\"\"\n",
-        "    X = np.random.rand(N)\n",
+        "    X = np.random.uniform(size=N)\n",
         "    Y_clean = compute_polynomial(X, theta)\n",
         "    Y = Y_clean + np.random.randn(N) * sigma\n",
         "    return X,Y\n",