A small pythonic change