Did you copy that code from an old version of the notebook from someplace? That is a very different definition than what I showed, so there is an obvious incompatibility here. If you got a solution from GitHub, please realize that a) that is cheating and b) it doesn’t work if the solution is too old. The assignments in most of the courses here change over time.
I suggest you get a clean copy of the assignment and start over.