correct the return type