Fix bad return method