Updating tests for hook features