Fix logic (wtf, good morning myself)