Fixing error

And I'm going to bed