Fix Python imports