Change hooks and its tests to work with work_in() change.