fix shape issue