fixed one broken test