Fixing Broken test