diff --git a/mitdeeplearning/lab1.py b/mitdeeplearning/lab1.py
index 6e66b926..02d566a9 100644
--- a/mitdeeplearning/lab1.py
+++ b/mitdeeplearning/lab1.py
@@ -81,7 +81,7 @@ def test_batch_func_next_step(func, args):
 
 def test_custom_dense_layer_output(y):
     # define the ground truth value for the array
-    true_y = np.array([[0.27064407,  0.1826951,  0.50374055]],dtype='float32')
+    true_y = np.array([[0.26978594 0.45750412 0.66536945]],dtype='float32')
     assert tf.shape(y).numpy().tolist() == list(true_y.shape), "[FAIL] output is of incorrect shape. expected {} but got {}".format(true_y.shape, y.numpy().shape)
     np.testing.assert_almost_equal(y.numpy(), true_y, decimal=7, err_msg="[FAIL] output is of incorrect value. expected {} but got {}".format(true_y, y.numpy()), verbose=True)
     print("[PASS] test_custom_dense_layer_output")