1 files changed, 7 insertions, 3 deletions
diff --git a/test/drifting_clock.h b/test/drifting_clock.h
index 2539b61786..3471c008a1 100644
--- a/test/drifting_clock.h
+++ b/test/drifting_clock.h
@@ -30,12 +30,16 @@ class DriftingClock : public Clock {
     return 1.0f - percent / 100.0f;
   }
 
-  Timestamp CurrentTime() override;
-  NtpTime CurrentNtpTime() override;
-  int64_t CurrentNtpInMilliseconds() override;
+  Timestamp CurrentTime() override { return Drift(clock_->CurrentTime()); }
+  NtpTime CurrentNtpTime() override { return Drift(clock_->CurrentNtpTime()); }
+  NtpTime ConvertTimestampToNtpTime(Timestamp timestamp) override {
+    return Drift(clock_->ConvertTimestampToNtpTime(timestamp));
+  }
 
  private:
   TimeDelta Drift() const;
+  Timestamp Drift(Timestamp timestamp) const;
+  NtpTime Drift(NtpTime ntp_time) const;
 
   Clock* const clock_;
   const float drift_;