Fix logic

Stupid code