Update tests for hooks