diff --git a/TODO.md b/TODO.md
index 3a06830..0e304f7 100644
--- a/TODO.md
+++ b/TODO.md
@@ -16,4 +16,5 @@
 8. Keybinds
 9. nth derivative support (again)
 10. add configs for toggling display of roots and extrema
-11. reduce jittering of roots and extrema points
\ No newline at end of file
+11. reduce jittering of roots and extrema points
+12. Update function tests
\ No newline at end of file
diff --git a/src/function.rs b/src/function.rs
index a735f37..3f9f673 100644
--- a/src/function.rs
+++ b/src/function.rs
@@ -355,7 +355,7 @@ fn verify_function(
 ) {
     {
         let (back_values, bars, derivative) = function.run_back();
-        assert!(derivative.is_none());
+        assert!(derivative.is_some());
         assert!(bars.is_none());
         assert_eq!(back_values.len(), pixel_width);
         let back_values_tuple: Vec<(f64, f64)> =
@@ -366,7 +366,7 @@ fn verify_function(
     {
         *function = function.clone().integral(true);
         let (back_values, bars, derivative) = function.run_back();
-        assert!(derivative.is_none());
+        assert!(derivative.is_some());
         assert!(bars.is_some());
         assert_eq!(back_values.len(), pixel_width);
 
@@ -381,7 +381,6 @@ fn verify_function(
     }
 
     {
-        function.derivative = true;
         let (back_values, bars, derivative) = function.run_back();
         assert!(derivative.is_some());