Fix weird return statement