Fix for higlighting the correct residues

I have no idea how many times I've "fixed" this code here.