add test hooks