Fix breakage with < Python 3