Fixing broken test