diff --git a/govarnam/govarnam_ml_test.go b/govarnam/govarnam_ml_test.go
index ae32a7b..877de53 100644
--- a/govarnam/govarnam_ml_test.go
+++ b/govarnam/govarnam_ml_test.go
@@ -3,6 +3,7 @@ package govarnam
 import (
 	"context"
 	"log"
+	"os"
 	"path"
 	"strings"
 	"testing"
diff --git a/govarnam/govarnam_test.go b/govarnam/govarnam_test.go
index 06b6b67..6400584 100644
--- a/govarnam/govarnam_test.go
+++ b/govarnam/govarnam_test.go
@@ -103,7 +103,7 @@ func TestMain(m *testing.M) {
 		log.Fatal(err)
 	}
 
-	testTempDir, err = os.TempDir("", "govarnam_test")
+	testTempDir, err = os.MkdirTemp("", "govarnam_test")
 	checkError(err)
 
 	for _, schemeDetail := range schemeDetails {