fixing edge case scenario