fix imports for generated stepDefinition