fixing one liner if