Bug fix

Hopefully