integrationtest: Add missing import