diff --git a/lib/pipely/component.rb b/lib/pipely/component.rb
index 4879274..3d1d5f2 100644
--- a/lib/pipely/component.rb
+++ b/lib/pipely/component.rb
@@ -28,7 +28,7 @@ class Component
       'FAILED' => 'orangered',
     }
 
-    include Virtus.model
+    include Virtus
 
     attribute :id, String
     attribute :type, String
diff --git a/pipely.gemspec b/pipely.gemspec
index a6dae68..e90fa16 100644
--- a/pipely.gemspec
+++ b/pipely.gemspec
@@ -16,7 +16,7 @@ Gem::Specification.new do |s|
 
   s.add_dependency "ruby-graphviz"
   s.add_dependency "rake"
-  s.add_dependency "virtus", "~>1.0.0"
+  s.add_dependency "virtus", "~>0.5.5"
   s.add_dependency "fog", "~>1.23"
   s.add_dependency "aws-sdk", "~>1.48"
   s.add_dependency "unf"