Fix import failure in mock test
1 file changed