Minor python fix