diff --git a/src/Orchard.Tests/Environment/DefaultCompositionStrategyTests.cs b/src/Orchard.Tests/Environment/DefaultCompositionStrategyTests.cs
index 60985cf75..63522039a 100644
--- a/src/Orchard.Tests/Environment/DefaultCompositionStrategyTests.cs
+++ b/src/Orchard.Tests/Environment/DefaultCompositionStrategyTests.cs
@@ -40,6 +40,10 @@ namespace Orchard.Tests.Environment {
                 .Setup(x => x.AvailableExtensions())
                 .Returns(() => _extensionDescriptors);
 
+            _container.Mock<IExtensionManager>()
+                .Setup(x => x.AvailableFeatures())
+                .Returns(() => _extensionDescriptors.SelectMany(ext => ext.Features));
+
             _container.Mock<IExtensionManager>()
                 .Setup(x => x.LoadFeatures(It.IsAny<IEnumerable<FeatureDescriptor>>()))
                 .Returns((IEnumerable<FeatureDescriptor> x) => StubLoadFeatures(x));