Updated imports in smoketest prod