Minor. Reverted optimized import in generated test