fix in BatchDecomposition