fix: proper return