diff --git a/tests/src/Util/Logger/ProfilerLoggerTest.php b/tests/src/Util/Logger/ProfilerLoggerTest.php
index 0a63b4381..2dbb604f9 100644
--- a/tests/src/Util/Logger/ProfilerLoggerTest.php
+++ b/tests/src/Util/Logger/ProfilerLoggerTest.php
@@ -59,9 +59,9 @@ class ProfilerLoggerTest extends MockedTest
 		$logger = new ProfilerLogger($this->logger, $this->profiler);
 
 		$this->logger->shouldReceive($function)->with($message, $context)->once();
-		$this->profiler->shouldReceive('startRecording');
+		$this->profiler->shouldReceive('startRecording')->with(\Mockery::any(), 'file')->once();
 		$this->profiler->shouldReceive('stopRecording');
-		$this->profiler->shouldReceive('saveTimestamp')->with(\Mockery::any(), 'file')->once();
+		$this->profiler->shouldReceive('saveTimestamp');
 		$logger->$function($message, $context);
 	}
 
@@ -74,9 +74,9 @@ class ProfilerLoggerTest extends MockedTest
 		$logger = new ProfilerLogger($this->logger, $this->profiler);
 
 		$this->logger->shouldReceive('log')->with(LogLevel::WARNING, 'test', ['a' => 'context'])->once();
-		$this->profiler->shouldReceive('startRecording');
+		$this->profiler->shouldReceive('startRecording')->with(\Mockery::any(), 'file')->once();
 		$this->profiler->shouldReceive('stopRecording');
-		$this->profiler->shouldReceive('saveTimestamp')->with(\Mockery::any(), 'file')->once();
+		$this->profiler->shouldReceive('saveTimestamp');
 
 		$logger->log(LogLevel::WARNING, 'test', ['a' => 'context']);
 	}