lib/gooddata/lcm/actions/collect_segment_clients.rb in gooddata-0.6.53 vs lib/gooddata/lcm/actions/collect_segment_clients.rb in gooddata-0.6.54

@@ -41,10 +41,11 @@
         def call(params)
           client = params.gdc_gd_client
 
           domain_name = params.organization || params.domain
           domain = client.domain(domain_name) || fail("Invalid domain name specified - #{domain_name}")
-          domain_segments = domain.segments
+          data_product = params.data_product
+          domain_segments = domain.segments(:all, data_product)
 
           segments = params.segments.map do |seg|
             domain_segments.find do |s|
               s.segment_id == seg.segment_id
             end