diff --git a/graphene_federation/service.py b/graphene_federation/service.py index 86b75b0..7295a17 100644 --- a/graphene_federation/service.py +++ b/graphene_federation/service.py @@ -26,7 +26,7 @@ def get_sdl(schema, custom_entities): string_schema = pattern.sub(" ", string_schema) for entity_name, entity in custom_entities.items(): - type_def_re = r"(type %s[^\{]*)" % entity_name + type_def_re = r"(type %s [^\{]*)" % entity_name repl_str = r"\1 %s " % entity._sdl pattern = re.compile(type_def_re) string_schema = pattern.sub(repl_str, string_schema)