Some small logic mistakes