diff --git a/fsm_test.go b/fsm_test.go
index 9b1a5d3e8..490b01771 100644
--- a/fsm_test.go
+++ b/fsm_test.go
@@ -102,5 +102,5 @@ func TestPlanCommittingHandlesSectorCommitFailed(t *testing.T) {
 
 	require.NoError(t, planCommitting(events, m.state))
 
-	require.Equal(t, api.SectorStates[api.CommitFailed], api.SectorStates[m.state.State])
+	require.Equal(t, api.CommitFailed, m.state.State)
 }
diff --git a/types_test.go b/types_test.go
index f3ec17aa3..93bffc20c 100644
--- a/types_test.go
+++ b/types_test.go
@@ -18,7 +18,7 @@ func TestSectorInfoSelialization(t *testing.T) {
 	dummyCid := builtin.AccountActorCodeID
 
 	si := &SectorInfo{
-		State:    123,
+		State:    "stateful",
 		SectorID: 234,
 		Nonce:    345,
 		Pieces: []Piece{{