fixed the one broken test