Fix import in testsuite