Fixed issue with one-lined if statements