Fix bug in index plotting code for single model case.