Added a small check to StateSpaceTests