Fix logic fuckup