Correctly check dict index to be in us: was working already but now it's
more pythonic