diff --git a/engine/engine.go b/engine/engine.go
index 5fe0977..e66dc0c 100644
--- a/engine/engine.go
+++ b/engine/engine.go
@@ -83,7 +83,12 @@ func TestEngine(sessionId string) (engine.Engine, func()) {
 	en = en.WithFirst(hl.Init)
 
 	cleanFn := func() {
-		err := menuStorageService.Close()
+		err := en.Finish()
+		if err != nil {
+			logg.Errorf(err.Error())
+		}
+
+		err = menuStorageService.Close()
 		if err != nil {
 			logg.Errorf(err.Error())
 		}