Rakefile in middleman-remover-1.0.1 vs Rakefile in middleman-remover-1.1.0

@@ -1,2 +1,15 @@
 require "bundler/gem_tasks"
+require "middleman-core/version"
 
+task default: :test
+
+desc "test command"
+task :test do
+  version = ::Middleman::VERSION
+
+  if /^4\./ =~ ::Middleman::VERSION
+    sh 'cucumber --tags ~@v3 features/'
+  else
+    sh 'cucumber --tags ~@v4 features/'
+  end
+end