Fixing python export