I r stupid, fix logic