lib/meducation_sdk/helpers.rb in meducation_sdk-1.6.16 vs lib/meducation_sdk/helpers.rb in meducation_sdk-2.0.0

@@ -13,9 +13,10 @@
         "MeducationSDK::#{spi_type.gsub("::", "")}"
       end
     end
 
     def spi_type_for(sdk_type)
+      sdk_type = sdk_type.gsub(/Mock$/, "")
       if SDK_TO_SPI_MAPPINGS.has_key?(sdk_type)
         SDK_TO_SPI_MAPPINGS[sdk_type]
       else
         sdk_type.gsub("MeducationSDK::", "")
       end