diff --git a/src/main/java/ca/joeltherrien/randomforest/NumericCovariate.java b/src/main/java/ca/joeltherrien/randomforest/NumericCovariate.java
index 76c83b3..5d1c8dc 100644
--- a/src/main/java/ca/joeltherrien/randomforest/NumericCovariate.java
+++ b/src/main/java/ca/joeltherrien/randomforest/NumericCovariate.java
@@ -60,7 +60,7 @@ public class NumericCovariate implements Covariate<Double>{
         }
 
         @Override
-        public Covariate<Double> getParent() {
+        public NumericCovariate getParent() {
             return NumericCovariate.this;
         }
 
@@ -84,7 +84,7 @@ public class NumericCovariate implements Covariate<Double>{
         }
 
         @Override
-        public Covariate<Double> getParent() {
+        public NumericCovariate getParent() {
             return NumericCovariate.this;
         }