diff --git a/test/test_version_class.py b/test/test_version_class.py
index 9607ab4..881ea7f 100644
--- a/test/test_version_class.py
+++ b/test/test_version_class.py
@@ -22,7 +22,7 @@ from version_repository import VersionRepository
 from release_type import ReleaseType    
 
 def test_version():
-    version = Version([1, 2, 3])
+    version = Version([1, 2, 3], ReleaseType.SNAPSHOT)
 
     version.increment(ReleaseType.SNAPSHOT)
     assert version.get_version_string() == "1.2.3-SNAPSHOT"
@@ -62,8 +62,9 @@ def test_gradle(tmp_path):
 
     # test
     version_repo = VersionRepository.get(f)
-    version = Version.create_release_version(ReleaseType.SNAPSHOT)
-    version_repo.write_file(version.get_version_string)    
+    version_list = version_repo.version_list
+    version = Version.create_release_version(version_list, ReleaseType.SNAPSHOT)
+    version_repo.write_file(version.get_version_string())    
     
     # check
     assert 'version = "12.4.678-SNAPSHOT"' in f.read_text()