A small pythonic change :)