Missing return of class