fixed small edge case