Fix logic as I bloody broke it