YH-25: Testing fixes