update for new hooks in 0.6.0