Fix a bug in an edge case