diff --git a/api/src/test/java/org/openmrs/module/fhir2/api/translators/impl/ValueSetTranslatorImplTest.java b/api/src/test/java/org/openmrs/module/fhir2/api/translators/impl/ValueSetTranslatorImplTest.java
index 799406591..feaa333e1 100644
--- a/api/src/test/java/org/openmrs/module/fhir2/api/translators/impl/ValueSetTranslatorImplTest.java
+++ b/api/src/test/java/org/openmrs/module/fhir2/api/translators/impl/ValueSetTranslatorImplTest.java
@@ -33,6 +33,7 @@
 import org.openmrs.ConceptReferenceTerm;
 import org.openmrs.ConceptSet;
 import org.openmrs.ConceptSource;
+import org.openmrs.ConceptMapType;
 import org.openmrs.module.fhir2.FhirTestConstants;
 import org.openmrs.module.fhir2.TestFhirSpringConfiguration;
 import org.openmrs.module.fhir2.api.FhirConceptSourceService;
@@ -75,8 +76,10 @@ public void shouldTranslateConceptSetToValueSet() {
 		ConceptMap conceptMap = mock(ConceptMap.class);
 		ConceptReferenceTerm conceptReferenceTerm = mock(ConceptReferenceTerm.class);
 		ConceptSource conceptSource = mock(ConceptSource.class);
+		ConceptMapType conceptMapType=mock(ConceptMapType.class);
 		
 		when(conceptMap.getConceptReferenceTerm()).thenReturn(conceptReferenceTerm);
+		when(conceptMap.getConceptMapType()).thenReturn(conceptMapType);
 		when(conceptReferenceTerm.getConceptSource()).thenReturn(conceptSource);
 		when(conceptReferenceTerm.getCode()).thenReturn("1000-1");
 		when(conceptSource.getName()).thenReturn("LOINC");