Fix failing python style test