1 files changed, 3 insertions, 2 deletions

diff --git a/subprojects/language/src/testFixtures/java/tools/refinery/language/model/tests/utils/WrappedProblem.java b/subprojects/language/src/testFixtures/java/tools/refinery/language/model/tests/utils/WrappedProblem.java
index fc51ff57..58bfce44 100644
--- a/subprojects/language/src/testFixtures/java/tools/refinery/language/model/tests/utils/WrappedProblem.java
+++ b/subprojects/language/src/testFixtures/java/tools/refinery/language/model/tests/utils/WrappedProblem.java
@@ -67,8 +67,9 @@ public record WrappedProblem(Problem problem) {
                 return ProblemNavigationUtil.named(problem.getNodes(), name);
         }
 
-        public Node individualNode(String name) {
-                var uniqueNodes = statementsOfType(IndividualDeclaration.class)
+        public Node atomNode(String name) {
+                var uniqueNodes = statementsOfType(NodeDeclaration.class)
+                                .filter(declaration -> declaration.getKind() == NodeKind.ATOM)
                                 .flatMap(declaration -> declaration.getNodes().stream());
                 return ProblemNavigationUtil.named(uniqueNodes, name);
         }