Whoops, add missing 'then' in if statement.