From 66daa03faf795eebd1d66be608b2507b34367b1b Mon Sep 17 00:00:00 2001
From: Reece Adamson <radamson@mitre.org>
Date: Thu, 11 Apr 2019 14:46:59 -0400
Subject: [PATCH] update versioned_resource_class

---
 lib/fhir_client/version_management.rb | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/lib/fhir_client/version_management.rb b/lib/fhir_client/version_management.rb
index a78887d1..bb584083 100644
--- a/lib/fhir_client/version_management.rb
+++ b/lib/fhir_client/version_management.rb
@@ -2,12 +2,16 @@ module FHIR
   module VersionManagement
 
     def versioned_resource_class(klass = nil)
-      return FHIR if klass.nil?
-      if @fhir_version == :stu3
-        FHIR::STU3.const_get(klass)
-      else
-        FHIR::DSTU2.const_get(klass)
-      end
+      mod = case @fhir_version
+            when :stu3
+              FHIR::STU3
+            when :dstu2
+              FHIR::DSTU2
+            else
+              FHIR
+            end
+      return mod if klass.nil?
+      mod.const_get(klass)
     end
 
     def versioned_format_class(format = nil)