fix wrong shape