fixed problem with the return types
fixed problem with the return types