Update another import for the new test runner [MJ-818]