Fix bad return