Small python fix.