Ensure the hook functions work properly