Bug fix

Pjotr messed up the code. Here's the fix. Thank me later