Fixing python issue (Thanks Dale)