From 9da94fcb50283cfc9cc949b4ca2d2c28a07c593d Mon Sep 17 00:00:00 2001
From: Slobodan Ilic <slobodan@crunch.io>
Date: Tue, 26 Jun 2018 13:25:12 +0200
Subject: [PATCH] [#158497157]: Fix test failures after rebase

---
 src/cr/cube/cube_slice.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/cr/cube/cube_slice.py b/src/cr/cube/cube_slice.py
index 7a06b5814..2267a9445 100644
--- a/src/cr/cube/cube_slice.py
+++ b/src/cr/cube/cube_slice.py
@@ -118,8 +118,10 @@ def _update_result(self, result):
     def _call_cube_method(self, method, *args, **kwargs):
         kwargs = self._update_args(kwargs)
         result = getattr(self._cube, method)(*args, **kwargs)
-        if method in ('labels', 'inserted_hs_indices') and not self.ca_as_0th:
-            return result[-2:]
+        if method in ('labels', 'inserted_hs_indices'):
+            if not self.ca_as_0th:
+                result = result[-2:]
+            return result
         return self._update_result(result)
 
     @property