Made a small change to make a line shorter (PEP8 compliant)