diff --git a/tests/unit/controller/VaultControllerTest.php b/tests/unit/controller/VaultControllerTest.php
index f94a7464..69a60a23 100644
--- a/tests/unit/controller/VaultControllerTest.php
+++ b/tests/unit/controller/VaultControllerTest.php
@@ -24,6 +24,7 @@
 namespace OCA\Passman\Controller;
 
 use OCA\Passman\Service\CredentialService;
+use OCA\Passman\Service\DeleteVaultRequestService;
 use OCA\Passman\Service\SettingsService;
 use OCA\Passman\Service\VaultService;
 use PHPUnit_Framework_TestCase;
@@ -42,15 +43,17 @@ class VaultControllerTest extends PHPUnit_Framework_TestCase {
 	private $userId = 'example';
 	private $credentialService;
 	private $vaultService;
+	private $deleteVaultRequestService;
 
 	public function setUp() {
 		$request = $this->getMockBuilder('OCP\IRequest')->getMock();
 		$this->vaultService = $this->createMock(VaultService::class);
 		$this->credentialService = $this->createMock(CredentialService::class);
 		$this->settingsService = $this->createMock(SettingsService::class);
+		$this->deleteVaultRequestService = $this->createMock(DeleteVaultRequestService::class);
 
 		$this->controller = new VaultController(
-			'passman', $request, $this->userId, $this->vaultService, $this->credentialService, $this->settingsService
+			'passman', $request, $this->userId, $this->vaultService, $this->credentialService, $this->deleteVaultRequestService,  $this->settingsService
 		);
 	}
 
diff --git a/tests/unit/lib/Db/CredentialMapperTest.php b/tests/unit/lib/Db/CredentialMapperTest.php
index b9b1d5c0..525d2961 100644
--- a/tests/unit/lib/Db/CredentialMapperTest.php
+++ b/tests/unit/lib/Db/CredentialMapperTest.php
@@ -245,7 +245,7 @@ class CredentialMapperTest extends DatabaseHelperTest {
 			'shared_key'		=> $original_row['shared_key'] . "asdf"
 		];
 
-		$updated = $this->mapper->updateCredential($raw_credential);
+		$updated = $this->mapper->updateCredential($raw_credential, false);
 		foreach ($raw_credential as $key => $value) {
 			if ($key === 'guid') continue;
 			$method = 'get' . str_replace('_', '', ucwords($key, '_'));