diff --git a/priorConditionedAnnealing/noise.py b/priorConditionedAnnealing/noise.py
index 5c53b9f..39ac441 100644
--- a/priorConditionedAnnealing/noise.py
+++ b/priorConditionedAnnealing/noise.py
@@ -22,7 +22,7 @@ class Colored_Noise():
 
     def reset(self, random_state=None):
         self.samples = cn.powerlaw_psd_gaussian(
-            self.beta, self.shape + (self.num_samples,), random_state=random_state)
+            self.beta, self.known_shape + (self.num_samples,), random_state=random_state)
 
 
 class Pink_Noise(Colored_Noise):
@@ -101,7 +101,7 @@ class Perlin_Noise():
     def __call__(self, shape):
         self.index += 1
         return [self.noise([self.index*self.scale, self.magic*(a+1)]) / self.normal_factor
-                for a in range(self.shape[-1])]
+                for a in range(self.known_shape[-1])]
 
     def reset(self):
         self.index = 0