Fixed incorrect return type