Fixed potential issue from n_chains not dividing evenly