diff --git a/Podfile.lock b/Podfile.lock
index 0a8a8432..b496552c 100644
--- a/Podfile.lock
+++ b/Podfile.lock
@@ -11,7 +11,7 @@ PODS:
 
 DEPENDENCIES:
   - Commander (~> 0.8)
-  - Kanna
+  - Kanna (~> 4.0.0)
   - PathKit (~> 0.9.0)
   - StencilSwiftKit (~> 2.6)
   - SwiftLint (~> 0.25)
diff --git a/Pods/Manifest.lock b/Pods/Manifest.lock
index 0a8a8432..b496552c 100644
--- a/Pods/Manifest.lock
+++ b/Pods/Manifest.lock
@@ -11,7 +11,7 @@ PODS:
 
 DEPENDENCIES:
   - Commander (~> 0.8)
-  - Kanna
+  - Kanna (~> 4.0.0)
   - PathKit (~> 0.9.0)
   - StencilSwiftKit (~> 2.6)
   - SwiftLint (~> 0.25)
diff --git a/SwiftGen.xcodeproj/xcshareddata/xcschemes/swiftgen.xcscheme b/SwiftGen.xcodeproj/xcshareddata/xcschemes/swiftgen.xcscheme
index 25212913..75183f80 100644
--- a/SwiftGen.xcodeproj/xcshareddata/xcschemes/swiftgen.xcscheme
+++ b/SwiftGen.xcodeproj/xcshareddata/xcschemes/swiftgen.xcscheme
@@ -138,6 +138,10 @@
             argument = "xcassets -t swift4 $(PROJECT_DIR)/Tests/Fixtures/Resources/XCAssets/Data.xcassets"
             isEnabled = "NO">
          </CommandLineArgument>
+         <CommandLineArgument
+            argument = "yaml $(PROJECT_DIR)/Tests/Fixtures/Resources/Yaml/root-mapping.yaml"
+            isEnabled = "NO">
+         </CommandLineArgument>
          <CommandLineArgument
             argument = "templates list"
             isEnabled = "NO">
diff --git a/SwiftGenKit.podspec b/SwiftGenKit.podspec
index 368bab12..914f36de 100644
--- a/SwiftGenKit.podspec
+++ b/SwiftGenKit.podspec
@@ -23,7 +23,7 @@ Pod::Spec.new do |s|
   s.source       = { git: 'https://github.com/SwiftGen/SwiftGenKit.git', tag: s.version.to_s }
   s.source_files = 'Sources/**/*.swift'
 
-  s.dependency 'Kanna'
+  s.dependency 'Kanna', '~> 4.0.0'
   s.dependency 'PathKit', '~> 0.9.0'
   s.dependency 'Yams', '~> 1.0.0'
   s.framework = 'Foundation'