diff --git a/client/state/db_test.go b/client/state/db_test.go
index 230c7ad86..c37ed8bd2 100644
--- a/client/state/db_test.go
+++ b/client/state/db_test.go
@@ -76,12 +76,6 @@ func TestStateDB_Allocations(t *testing.T) {
 		alloc1 := mock.Alloc()
 		alloc2 := mock.BatchAlloc()
 
-		//XXX Sadly roundtripping allocs loses time.Duration type
-		//    information from the Config map[string]interface{}. As
-		//    the mock driver itself with unmarshal run_for into the
-		//    proper type, we can safely ignore it here.
-		delete(alloc2.Job.TaskGroups[0].Tasks[0].Config, "run_for")
-
 		require.NoError(db.PutAllocation(alloc1))
 		require.NoError(db.PutAllocation(alloc2))