fix wip test