diff --git a/apps/settings/tests/Controller/CheckSetupControllerTest.php b/apps/settings/tests/Controller/CheckSetupControllerTest.php
index b13419c40d06f..dd7cf2de91f9b 100644
--- a/apps/settings/tests/Controller/CheckSetupControllerTest.php
+++ b/apps/settings/tests/Controller/CheckSetupControllerTest.php
@@ -47,6 +47,7 @@
 use OCP\AppFramework\Http\DataDisplayResponse;
 use OCP\AppFramework\Http\DataResponse;
 use OCP\AppFramework\Http\RedirectResponse;
+use OCP\DB\IResult;
 use OCP\DB\QueryBuilder\IFunctionBuilder;
 use OCP\DB\QueryBuilder\IQueryBuilder;
 use OCP\DB\QueryBuilder\IQueryFunction;
@@ -621,6 +622,10 @@ public function testCheck() {
 			->willReturn($functionBuilder);
 		$functionBuilder->method('count')
 			->willReturn($this->getMockBuilder(IQueryFunction::class)->getMock());
+		$queryBuilder->method('from')
+			->willReturn($queryBuilder);
+		$queryBuilder->method('executeQuery')
+			->willReturn($this->getMockBuilder(IResult::class)->getMock());
 
 		$expected = new DataResponse(
 			[