Fix minor logic fuckup