4 files changed, 12 insertions, 10 deletions
diff --git a/lang/c/pseudoc.cpp b/lang/c/pseudoc.cpp
index dff591ff..52622f57 100644
--- a/lang/c/pseudoc.cpp
+++ b/lang/c/pseudoc.cpp
@@ -6,9 +6,9 @@ using namespace std;
 using namespace BinaryNinja;
 
 
-PseudoCFunction::PseudoCFunction(
-	Architecture* arch, Function* owner, HighLevelILFunction* highLevelILFunction) :
-	LanguageRepresentationFunction(arch, owner, highLevelILFunction), m_highLevelIL(highLevelILFunction)
+PseudoCFunction::PseudoCFunction(LanguageRepresentationFunctionType* type, Architecture* arch, Function* owner,
+	HighLevelILFunction* highLevelILFunction) :
+	LanguageRepresentationFunction(type, arch, owner, highLevelILFunction), m_highLevelIL(highLevelILFunction)
 {
 }
 
@@ -2766,7 +2766,7 @@ PseudoCFunctionType::PseudoCFunctionType(): LanguageRepresentationFunctionType("
 Ref<LanguageRepresentationFunction> PseudoCFunctionType::Create(Architecture* arch, Function* owner,
 	HighLevelILFunction* highLevelILFunction)
 {
-	return new PseudoCFunction(arch, owner, highLevelILFunction);
+	return new PseudoCFunction(this, arch, owner, highLevelILFunction);
 }
 
 
diff --git a/lang/c/pseudoc.h b/lang/c/pseudoc.h
index 024be2f1..feb64552 100644
--- a/lang/c/pseudoc.h
+++ b/lang/c/pseudoc.h
@@ -51,7 +51,8 @@ protected:
 		const BinaryNinja::HighLevelILInstruction& instr, BinaryNinja::HighLevelILTokenEmitter& tokens) override;
 
 public:
-	PseudoCFunction(BinaryNinja::Architecture* arch, BinaryNinja::Function* owner, BinaryNinja::HighLevelILFunction* highLevelILFunction);
+	PseudoCFunction(BinaryNinja::LanguageRepresentationFunctionType* type, BinaryNinja::Architecture* arch,
+		BinaryNinja::Function* owner, BinaryNinja::HighLevelILFunction* highLevelILFunction);
 
 	std::string GetAnnotationStartString() const override;
 	std::string GetAnnotationEndString() const override;
diff --git a/lang/rust/pseudorust.cpp b/lang/rust/pseudorust.cpp
index 837a7932..a1f7defe 100644
--- a/lang/rust/pseudorust.cpp
+++ b/lang/rust/pseudorust.cpp
@@ -7,9 +7,9 @@ using namespace std;
 using namespace BinaryNinja;
 
 
-PseudoRustFunction::PseudoRustFunction(
-	Architecture* arch, Function* owner, HighLevelILFunction* highLevelILFunction) :
-	LanguageRepresentationFunction(arch, owner, highLevelILFunction), m_highLevelIL(highLevelILFunction)
+PseudoRustFunction::PseudoRustFunction(LanguageRepresentationFunctionType* type, Architecture* arch, Function* owner,
+	HighLevelILFunction* highLevelILFunction) :
+	LanguageRepresentationFunction(type, arch, owner, highLevelILFunction), m_highLevelIL(highLevelILFunction)
 {
 }
 
@@ -2814,7 +2814,7 @@ PseudoRustFunctionType::PseudoRustFunctionType(): LanguageRepresentationFunction
 Ref<LanguageRepresentationFunction> PseudoRustFunctionType::Create(Architecture* arch, Function* owner,
 	HighLevelILFunction* highLevelILFunction)
 {
-	return new PseudoRustFunction(arch, owner, highLevelILFunction);
+	return new PseudoRustFunction(this, arch, owner, highLevelILFunction);
 }
 
 
diff --git a/lang/rust/pseudorust.h b/lang/rust/pseudorust.h
index a2ab9584..96e656c2 100644
--- a/lang/rust/pseudorust.h
+++ b/lang/rust/pseudorust.h
@@ -59,7 +59,8 @@ protected:
 	virtual void EndLines(const BinaryNinja::HighLevelILInstruction& instr, BinaryNinja::HighLevelILTokenEmitter& tokens) override;
 
 public:
-	PseudoRustFunction(BinaryNinja::Architecture* arch, BinaryNinja::Function* owner, BinaryNinja::HighLevelILFunction* highLevelILFunction);
+	PseudoRustFunction(BinaryNinja::LanguageRepresentationFunctionType* type, BinaryNinja::Architecture* arch,
+		BinaryNinja::Function* owner, BinaryNinja::HighLevelILFunction* highLevelILFunction);
 
 	std::string GetAnnotationStartString() const override;
 	std::string GetAnnotationEndString() const override;