YH-471: Fixed minor bug