Fix typo
Unfinished logic that is more symmetric than original