diff --git a/run.c b/run.c
index 11af4dc..a77f5cd 100644
--- a/run.c
+++ b/run.c
@@ -13,7 +13,6 @@ $ ./run
 #include <time.h>
 #include <math.h>
 #include <string.h>
-#include <sys/time.h>
 
 // ----------------------------------------------------------------------------
 // Transformer and RunState structs, and related memory management
@@ -378,9 +377,9 @@ int argmax(float* v, int n) {
 // ----------------------------------------------------------------------------
 
 long time_in_ms() {
-  struct timeval time;
-  gettimeofday(&time, NULL);
-  return time.tv_sec * 1000 + time.tv_usec / 1000;
+  struct timespec time;
+  timespec_get(&time, TIME_UTC);
+  return time.tv_sec * 1000 + time.tv_nsec / 1000000;
 }
 
 int main(int argc, char *argv[]) {