fixing edge case