fix return type incorrect issue