Fixed bug with saving observations