diff --git a/tools/test/progress_mock.hpp b/tools/test/progress_mock.hpp
index 80c0af1..a43d9b3 100644
--- a/tools/test/progress_mock.hpp
+++ b/tools/test/progress_mock.hpp
@@ -12,8 +12,10 @@
 class ProgressMock : public ProgressInterface
 {
   public:
-    MOCK_METHOD1(updateProgress, void(std::int64_t));
-    MOCK_METHOD1(start, void(std::int64_t));
+    MOCK_METHOD(void, updateProgress, (std::int64_t), (override));
+    MOCK_METHOD(void, start, (std::int64_t), (override));
+    MOCK_METHOD(void, finish, (), (override));
+    MOCK_METHOD(void, abort, (), (override));
 };
 
 } // namespace host_tool