Fixing Broken Test