Fix import ambiguity (python side)