Bugfix

Forgot to change these lines earlier