Fix

- added missing return