fix a bug

thanks sanshi0518