Change the hook's return type