Readme.md in use_case-1.0.1 vs Readme.md in use_case-1.0.2

@@ -90,11 +90,11 @@
 #
 # This class uses Project.find to look up a project by id if project_id is
 # provided and project is not. This is the only class that directly touches
 # classes from the Rails application.
 class NewRepositoryInput
-  include Virtus
+  include Virtus.model
   attribute :name, String
   attribute :description, String
   attribute :project, Project
   attribute :project_id, Integer
 
@@ -341,10 +341,10 @@
     errors.add(:name, "is taken") if !uniq?
   end
 end
 
 class NewUserInput
-  include Virtus
+  include Virtus.model
   attribute :name, String
 end
 
 class NewUserCommand
   def execute(user)