Fixed incorrect return