Fixing missed single line if statements