fixed none expected shape saving issue