1 files changed, 6 insertions, 1 deletions
diff --git a/src/main/java/net/fabricmc/loom/configuration/DependencyProvider.java b/src/main/java/net/fabricmc/loom/configuration/DependencyProvider.java
index 9f6d6db3..5017d54a 100644
--- a/src/main/java/net/fabricmc/loom/configuration/DependencyProvider.java
+++ b/src/main/java/net/fabricmc/loom/configuration/DependencyProvider.java
@@ -49,6 +49,7 @@ import org.zeroturnaround.zip.ZipUtil;
 
 import net.fabricmc.loom.LoomGradleExtension;
 import net.fabricmc.loom.LoomGradlePlugin;
+import net.fabricmc.loom.extension.LoomFiles;
 
 public abstract class DependencyProvider {
 	private LoomDependencyManager dependencyManager;
@@ -57,7 +58,7 @@ public abstract class DependencyProvider {
 
 	public DependencyProvider(Project project) {
 		this.project = project;
-		this.extension = project.getExtensions().getByType(LoomGradleExtension.class);
+		this.extension = LoomGradleExtension.get(project);
 	}
 
 	public abstract void provide(DependencyInfo dependency, Consumer<Runnable> postPopulationScheduler) throws Exception;
@@ -88,6 +89,10 @@ public abstract class DependencyProvider {
 		return extension;
 	}
 
+	public LoomFiles getDirectories() {
+		return getExtension().getFiles();
+	}
+
 	public boolean isRefreshDeps() {
 		return LoomGradlePlugin.refreshDeps;
 	}