fixed another return type problem