diff --git a/internal/common/sleep_test.go b/internal/common/sleep_test.go
index 5ecb6ea..bba4661 100644
--- a/internal/common/sleep_test.go
+++ b/internal/common/sleep_test.go
@@ -2,6 +2,7 @@ package common_test
 
 import (
 	"context"
+	"errors"
 	"testing"
 	"time"
 
@@ -13,7 +14,7 @@ func TestSleep(test *testing.T) {
 	var t = func(name string, ctx context.Context, expected error) {
 		test.Run(name, func(test *testing.T) {
 			var err = common.Sleep(ctx, dt)
-			if err != expected {
+			if !errors.Is(err, expected) {
 				test.Errorf("expected %v, got %v", expected, err)
 			}
 		})
diff --git a/load/load_test.go b/load/load_test.go
index d2739ba..4c9483b 100644
--- a/load/load_test.go
+++ b/load/load_test.go
@@ -1,6 +1,7 @@
 package load
 
 import (
+	"errors"
 	"fmt"
 	"testing"
 
@@ -8,7 +9,7 @@ import (
 )
 
 func skipIfNotImplementedErr(t testing.TB, err error) {
-	if err == common.ErrNotImplementedError {
+	if errors.Is(err, common.ErrNotImplementedError) {
 		t.Skip("not implemented")
 	}
 }