fix minor issue in previous merge