Simplify the implementation a bit by unifying one corner case.