diff --git a/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java b/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java
index df478bec689e540671497a8fc80934fd5596b6b5..7cf2656b5b47026abb089690cd79433f16a844d4 100644
--- a/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java
+++ b/src/main/java/kieker/analysis/stage/tracediagnosis/TraceAggregationComposite.java
@@ -18,8 +18,8 @@ package kieker.analysis.stage.tracediagnosis;
 
 import java.util.List;
 
-import kieker.analysis.stage.graphoutput.DotGraphWriter;
-import kieker.analysis.stage.graphoutput.GraphMLWriter;
+import kieker.analysis.trace.graphoutput.DotGraphWriter;
+import kieker.analysis.trace.graphoutput.GraphMLWriter;
 import kieker.analysis.trace.traversal.AggrTraceTraverserStage;
 import kieker.analysis.trace.traversal.NamedGraph;
 import kieker.analysis.traceanalysisdomain.AggregatedTrace;
diff --git a/src/main/java/kieker/analysis/stage/graphoutput/DotGraphWriter.java b/src/main/java/kieker/analysis/trace/graphoutput/DotGraphWriter.java
similarity index 100%
rename from src/main/java/kieker/analysis/stage/graphoutput/DotGraphWriter.java
rename to src/main/java/kieker/analysis/trace/graphoutput/DotGraphWriter.java
diff --git a/src/main/java/kieker/analysis/stage/graphoutput/GraphMLWriter.java b/src/main/java/kieker/analysis/trace/graphoutput/GraphMLWriter.java
similarity index 100%
rename from src/main/java/kieker/analysis/stage/graphoutput/GraphMLWriter.java
rename to src/main/java/kieker/analysis/trace/graphoutput/GraphMLWriter.java