fix: return function type