diff --git a/dingo/PolytopeSampler.py b/dingo/PolytopeSampler.py
index 03b11c44..8d99967c 100644
--- a/dingo/PolytopeSampler.py
+++ b/dingo/PolytopeSampler.py
@@ -203,9 +203,10 @@ def generate_steady_states_no_multiphase(
         P = HPolytope(self._A, self._b)
 
         samples = P.generate_samples(method, n, burn_in, thinning, self._parameters["fast_computations"])
+        samples_T = samples.T
 
         steady_states = map_samples_to_steady_states(
-                samples, self._N, self._N_shift
+                samples_T, self._N, self._N_shift
             )
 
         return steady_states
@@ -263,7 +264,8 @@ def sample_from_polytope_no_multiphase(
         except ImportError as e:
             samples = P.generate_samples(method, n, burn_in, thinning, False)
 
-        return samples
+        samples_T = samples.T
+        return samples_T
 
     @staticmethod
     def round_polytope(