Fix last commit.

Turns out the last PR didn't work properly, this fixes it. (I had messed up when porting the changes back to Coffeescript.)