diff --git a/mlclass-ex3/mlclass-ex3/predict.m b/mlclass-ex3/mlclass-ex3/predict.m
index 3176635..aa05253 100644
--- a/mlclass-ex3/mlclass-ex3/predict.m
+++ b/mlclass-ex3/mlclass-ex3/predict.m
@@ -24,7 +24,7 @@
 
 a1 = [ones(m,1) X];
 z2 = a1 *Theta1';
-a2 = [ones(size(z2),1) sigmoid(z2)];
+a2 = [ones(m,1) sigmoid(z2)];
 z3 = a2*Theta2';
 a3 = sigmoid(z3);