lib/rollo/commands/service_cluster.rb in rollo-0.1.0 vs lib/rollo/commands/service_cluster.rb in rollo-0.2.0

@@ -30,11 +30,11 @@
         batch_size = options[:batch_size]
         service_start_wait_minutes = options[:startup_time]
         service_start_wait_seconds = 60 * service_start_wait_minutes
 
         service_cluster = service_cluster ||
-            Rollo::ServiceCluster.new(ecs_cluster_name, region)
+            Rollo::Model::ServiceCluster.new(ecs_cluster_name, region)
 
         say("Increasing service instance counts by #{batch_size}...")
         with_padding do
           service_cluster.with_replica_services do |on|
             on.start do |services|
@@ -89,10 +89,10 @@
           region, _, ecs_cluster_name,
               service_cluster = nil)
         batch_size = options[:batch_size]
 
         service_cluster = service_cluster ||
-            Rollo::ServiceCluster.new(ecs_cluster_name, region)
+            Rollo::Model::ServiceCluster.new(ecs_cluster_name, region)
 
         say("Decreasing service instance counts by #{batch_size}...")
         with_padding do
           service_cluster.with_replica_services do |on|
             on.each_service do |service|