If you don’t believe me, here are the instructions for getting a clean version of the notebook.
Of course also note that you are far from the first person to attempt this assignment. If that really was the way it was written, everyone would have had this problem and it would have been fixed by now.
Also note that your fix is not going to really work. The shape of b is independent of the number of samples, right?