Fixed a small edge case.