Fix missing import in test case