bug fixed

I'll fix the crapy tests later, but this fixes the bug