fix wrong shape message bug