Make this fix more Pythonic.