fix issuse #4