diff --git a/handler_mock_test.go b/handler_mock_test.go
index b323d61..ef6f8c6 100644
--- a/handler_mock_test.go
+++ b/handler_mock_test.go
@@ -95,6 +95,16 @@ func (_mr *_MockFullDataStoreRecorder) GetReader(arg0 interface{}) *gomock.Call
 	return _mr.mock.ctrl.RecordCall(_mr.mock, "GetReader", arg0)
 }
 
+func (_m *MockFullDataStore) FinishUpload(id string) error {
+	ret := _m.ctrl.Call(_m, "FinishUpload", id)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+func (_mr *_MockFullDataStoreRecorder) FinishUpload(arg0 interface{}) *gomock.Call {
+	return _mr.mock.ctrl.RecordCall(_mr.mock, "FinishUpload", arg0)
+}
+
 // Mock of Locker interface
 type MockLocker struct {
 	ctrl     *gomock.Controller
diff --git a/patch_test.go b/patch_test.go
index 25bf058..48c8598 100644
--- a/patch_test.go
+++ b/patch_test.go
@@ -22,6 +22,7 @@ func TestPatch(t *testing.T) {
 				Size:   10,
 			}, nil),
 			store.EXPECT().WriteChunk("yes", int64(5), NewReaderMatcher("hello")).Return(int64(5), nil),
+			store.EXPECT().FinishUpload("yes"),
 		)
 
 		handler, _ := NewHandler(Config{
@@ -59,7 +60,7 @@ func TestPatch(t *testing.T) {
 			store.EXPECT().GetInfo("yes").Return(FileInfo{
 				ID:     "yes",
 				Offset: 5,
-				Size:   10,
+				Size:   20,
 			}, nil),
 			store.EXPECT().WriteChunk("yes", int64(5), NewReaderMatcher("hello")).Return(int64(5), nil),
 		)
@@ -188,6 +189,7 @@ func TestPatch(t *testing.T) {
 				Size:   20,
 			}, nil),
 			store.EXPECT().WriteChunk("yes", int64(5), NewReaderMatcher("hellothisismore")).Return(int64(15), nil),
+			store.EXPECT().FinishUpload("yes"),
 		)
 
 		handler, _ := NewHandler(Config{
diff --git a/utils_test.go b/utils_test.go
index e1bad9b..83eb811 100644
--- a/utils_test.go
+++ b/utils_test.go
@@ -18,6 +18,7 @@ type FullDataStore interface {
 	tusd.TerminaterDataStore
 	tusd.ConcaterDataStore
 	tusd.GetReaderDataStore
+	tusd.FinisherDataStore
 }
 
 type Locker interface {