diff --git a/autograder-cmd/src/main/java/de/firemage/autograder/cmd/Application.java b/autograder-cmd/src/main/java/de/firemage/autograder/cmd/Application.java
index 81cd7d43..f6d4d12c 100644
--- a/autograder-cmd/src/main/java/de/firemage/autograder/cmd/Application.java
+++ b/autograder-cmd/src/main/java/de/firemage/autograder/cmd/Application.java
@@ -154,7 +154,7 @@ private void execute(
                             );
 
                             String result = "[%s]: %s - Found problem in '%s'%n".formatted(
-                                    problem.getProblemType(),
+                                    problem.getType(),
                                     problem.getCheckName(),
                                     position.toString()
                             );
@@ -253,7 +253,7 @@ private void printProblemsAsJson(Collection<? extends AbstractProblem> problems,
             String jsonOutput = mapper.writeValueAsString(problems.stream().map(problem -> {
                 AbstractCodePosition position = problem.getPosition();
                 return new Annotation(
-                        problem.getProblemType(),
+                        problem.getType(),
                         linter.translateMessage(problem.getExplanation()),
                         position.path().toString().replace("\\", "/"),
                         position.startLine(),
diff --git a/autograder-cmd/src/main/java/de/firemage/autograder/cmd/output/Annotation.java b/autograder-cmd/src/main/java/de/firemage/autograder/cmd/output/Annotation.java
index 5aa2ecb8..b3316ade 100644
--- a/autograder-cmd/src/main/java/de/firemage/autograder/cmd/output/Annotation.java
+++ b/autograder-cmd/src/main/java/de/firemage/autograder/cmd/output/Annotation.java
@@ -1,6 +1,4 @@
 package de.firemage.autograder.cmd.output;
 
-import de.firemage.autograder.core.ProblemType;
-
-public record Annotation(ProblemType type, String message, String file, int startLine, int endLine) {
+public record Annotation(String type, String message, String file, int startLine, int endLine) {
 }