diff --git a/src/ObjectPool/test/DefaultObjectPoolTest.cs b/src/ObjectPool/test/DefaultObjectPoolTest.cs
index b44aa7e1c7..9bf84dca03 100644
--- a/src/ObjectPool/test/DefaultObjectPoolTest.cs
+++ b/src/ObjectPool/test/DefaultObjectPoolTest.cs
@@ -45,7 +45,7 @@ namespace Microsoft.Extensions.ObjectPool
         {
             // Arrange
             var pool = new DefaultObjectPool<List<int>>(new ListPolicy());
-            
+
             // Act
             var list = pool.Get();
 
diff --git a/src/Testing/src/xunit/FlakyAttribute.cs b/src/Testing/src/xunit/FlakyAttribute.cs
index f58026c7ca..ab4450e685 100644
--- a/src/Testing/src/xunit/FlakyAttribute.cs
+++ b/src/Testing/src/xunit/FlakyAttribute.cs
@@ -50,7 +50,7 @@ namespace Microsoft.AspNetCore.Testing.xunit
     /// </para>
     /// </example>
     [TraitDiscoverer("Microsoft.AspNetCore.Testing.xunit.FlakyTestDiscoverer", "Microsoft.AspNetCore.Testing")]
-    [AttributeUsage(AttributeTargets.Method)]
+    [AttributeUsage(AttributeTargets.Method | AttributeTargets.Class | AttributeTargets.Assembly)]
     public sealed class FlakyAttribute : Attribute, ITraitAttribute
     {
         /// <summary>