Small update to test hook