From 15a8a270d452f5b9423726f06f68b1d01e244afe Mon Sep 17 00:00:00 2001
From: Arno Solin <arno.solin@aalto.fi>
Date: Wed, 13 Nov 2013 18:14:11 +0000
Subject: [PATCH] Small fixes.

---
 GPy/models/state_space.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GPy/models/state_space.py b/GPy/models/state_space.py
index b3b70614..833af1fd 100644
--- a/GPy/models/state_space.py
+++ b/GPy/models/state_space.py
@@ -196,7 +196,7 @@ class StateSpace(Model):
         Y = self.posterior_samples_f(X,size)
 
         # Add noise
-        Y += np.sqrt(self.sigma2)*np.random.randn(Y.shape)
+        Y += np.sqrt(self.sigma2)*np.random.randn(Y.shape[0],Y.shape[1])
 
         # Return trajectory
         return Y