Fixed spmm issue