diff --git a/src/UQpy/stochastic_process/KarhunenLoeveExpansion.py b/src/UQpy/stochastic_process/KarhunenLoeveExpansion.py
index 35e1e6b27..51542d9cf 100644
--- a/src/UQpy/stochastic_process/KarhunenLoeveExpansion.py
+++ b/src/UQpy/stochastic_process/KarhunenLoeveExpansion.py
@@ -184,7 +184,7 @@ def run(self, nsamples, random_variables=None):
         The :meth:`run` method has no returns, although it creates and/or appends the :py:attr:`samples` attribute of
         the :class:`KarhunenLoeveExpansion` class.
         """
-        samples = np.zeros((nsamples, self.correlation_function.shape[0], self.correlation_function.shape[1]))
+        samples = np.zeros((nsamples, self.correlation_function.shape[0], self.correlation_function.shape[2]))
         if random_variables is None:
             random_variables = np.random.normal(size=[self.thresholds[1], self.thresholds[0], nsamples])
         else: