1 files changed, 6 insertions, 1 deletions

diff --git a/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQuery.java b/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQuery.java
index aaebfcc2..bf7651ad 100644
--- a/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQuery.java
+++ b/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQuery.java
@@ -55,10 +55,15 @@ public final class FunctionalQuery<T> extends Query<T> {
         }
 
         @Override
-        protected Query<T> withDnfInternal(Dnf newDnf) {
+        protected FunctionalQuery<T> withDnfInternal(Dnf newDnf) {
                 return newDnf.asFunction(type);
         }
 
+        @Override
+        public FunctionalQuery<T> withDnf(Dnf newDnf) {
+                return (FunctionalQuery<T>) super.withDnf(newDnf);
+        }
+
         public AssignedValue<T> call(List<NodeVariable> arguments) {
                 return targetVariable -> {
                         var argumentsWithTarget = new ArrayList<Variable>(arguments.size() + 1);