1 files changed, 2 insertions, 2 deletions
diff --git a/type.cpp b/type.cpp
index 12973b2a..50766dc8 100644
--- a/type.cpp
+++ b/type.cpp
@@ -2559,10 +2559,10 @@ vector<EnumerationMember> Enumeration::GetMembers() const
 }
 
 
-vector<InstructionTextToken> Enumeration::GetTokensForValue(uint64_t value, size_t width)
+vector<InstructionTextToken> Enumeration::GetTokensForValue(uint64_t value, size_t width, Ref<Type> type)
 {
 	size_t count;
-	BNInstructionTextToken* tokens = BNGetEnumerationTokensForValue(m_object, value, width, &count);
+	BNInstructionTextToken* tokens = BNGetEnumerationTokensForValue(m_object, value, width, &count, type->GetObject());
 
 	return InstructionTextToken::ConvertAndFreeInstructionTextTokenList(tokens, count);
 }