changes to prevent braking my code

I dont know if this was changed somewhere down the road, but i got a not
found error on line 41.
This seems to fix that.