fixing broken test