diff --git a/formula-android/src/test/java/com/instacart/formula/MockitoFormulaTest.kt b/formula-android/src/test/java/com/instacart/formula/MockitoFormulaTest.kt
index e611ce95..3aba8e80 100644
--- a/formula-android/src/test/java/com/instacart/formula/MockitoFormulaTest.kt
+++ b/formula-android/src/test/java/com/instacart/formula/MockitoFormulaTest.kt
@@ -16,9 +16,7 @@ class MockitoFormulaTest {
         assertThat(formula.type()).isEqualTo(MyFormula::class)
     }
 
-    class MyFormula : Formula<Unit, Unit, Unit>() {
-        override fun initialState(input: Unit) = Unit
-
+    class MyFormula : StatelessFormula<Unit, Unit>() {
         override fun Snapshot<Unit, Unit>.evaluate(): Evaluation<Unit> {
             return Evaluation(Unit)
         }
diff --git a/formula/src/main/java/com/instacart/formula/StatelessFormula.kt b/formula/src/main/java/com/instacart/formula/StatelessFormula.kt
index 2a8bba7f..164cfacd 100644
--- a/formula/src/main/java/com/instacart/formula/StatelessFormula.kt
+++ b/formula/src/main/java/com/instacart/formula/StatelessFormula.kt
@@ -13,7 +13,7 @@ package com.instacart.formula
 abstract class StatelessFormula<Input, Output> : IFormula<Input, Output> {
 
     // Implements the common API used by the runtime.
-    private val implementation = object : Formula<Input, Unit, Output>() {
+    private val implementation: Formula<Input, Unit, Output> = object : Formula<Input, Unit, Output>() {
         override fun initialState(input: Input) = Unit
 
         override fun Snapshot<Input, Unit>.evaluate(): Evaluation<Output> {