Fixed f*cking bug

So hard...