small python fix