diff --git a/types/proposal_test.go b/types/proposal_test.go
index a7a6083f8..8885a0bee 100644
--- a/types/proposal_test.go
+++ b/types/proposal_test.go
@@ -250,7 +250,7 @@ func TestIsTimely(t *testing.T) {
 			mockSource := new(tmtimemocks.Source)
 			mockSource.On("Now").Return(testCase.localTime)
 
-			ti := p.IsTimely(mockSource, tp)
+			ti := p.IsTimely(mockSource, tp, 1)
 			assert.Equal(t, testCase.expectTimely, ti)
 		})
 	}