Bug fix

Awww yeahhhh