Add test hook