small python syntax fix