BUG: shape bug in tests