diff --git a/trustgraph-flow/trustgraph/extract/kg/relationships/extract.py b/trustgraph-flow/trustgraph/extract/kg/relationships/extract.py
index dab6fb86..ac2929a3 100755
--- a/trustgraph-flow/trustgraph/extract/kg/relationships/extract.py
+++ b/trustgraph-flow/trustgraph/extract/kg/relationships/extract.py
@@ -150,7 +150,7 @@ class Processor(FlowProcessor):
                     o=Value(value=str(p), is_uri=False)
                 ))
 
-                if rel["o_entity"]:
+                if rel["object-entity"]:
                     # Label for o
                     triples.append(Triple(
                         s=o_value,
@@ -165,7 +165,7 @@ class Processor(FlowProcessor):
                     o=Value(value=v.metadata.id, is_uri=True)
                 ))
 
-                if rel["o_entity"]:
+                if rel["object-entity"]:
                     # 'Subject of' for o
                     triples.append(Triple(
                         s=o_value,
diff --git a/trustgraph-flow/trustgraph/model/prompt/template/service.py b/trustgraph-flow/trustgraph/model/prompt/template/service.py
index 3435b94c..67590c1c 100755
--- a/trustgraph-flow/trustgraph/model/prompt/template/service.py
+++ b/trustgraph-flow/trustgraph/model/prompt/template/service.py
@@ -13,7 +13,7 @@ from .... schema import PromptRequest, PromptResponse, Error
 from .... schema import TextCompletionRequest, TextCompletionResponse
 
 from .... base import FlowProcessor
-from .... base import ProducerSpec, ConsumerSpec, TextCompletionSpec
+from .... base import ProducerSpec, ConsumerSpec, TextCompletionClientSpec
 
 from . prompt_manager import PromptConfiguration, Prompt, PromptManager
 
@@ -43,7 +43,7 @@ class Processor(FlowProcessor):
         )
 
         self.register_specification(
-            TextCompletionSpec(
+            TextCompletionClientSpec(
                 request_name = "text-completion-request",
                 response_name = "text-completion-response",
             )