Fixed small python error