Fix some python