Moved fix over to coffeescript as well

Realized that I had only made the previous fix in JS and it would be blown away.