fix edge case