Test says that the space does not contain the generated state. Need to
verify why and probably reshape the observations if necessary.