fix task model test