Fix imports in DummyGenerator