Fixed the weird edge case.