[#1332] Fix bad import in test.