fixed bad return type