- fix return type