[JY] fix bug