Fixed bugs, YH