add test hook to demo