another small Pythonic change