Python coding style fix