fixed a small mistake in python script