Fixed another edge case