please use braces or do the same thing above (i.e., if + return)