Fix TaskViewModel