diff --git a/solver/llbsolver/metrics_test.go b/solver/llbsolver/metrics_test.go
index 659f27dd9..7aa008c56 100644
--- a/solver/llbsolver/metrics_test.go
+++ b/solver/llbsolver/metrics_test.go
@@ -1,6 +1,7 @@
 package llbsolver
 
 import (
+	"context"
 	"testing"
 	"time"