diff --git a/gpu/gpu_test.go b/gpu/gpu_test.go
index c260211e..b65bce81 100644
--- a/gpu/gpu_test.go
+++ b/gpu/gpu_test.go
@@ -18,7 +18,7 @@ func TestBasicGetGPUInfo(t *testing.T) {
 	case "linux", "windows":
 		assert.Greater(t, info.TotalMemory, uint64(0))
 		assert.Greater(t, info.FreeMemory, uint64(0))
-		assert.Greater(t, info.DeviceCount, uint64(0))
+		assert.Greater(t, info.DeviceCount, uint32(0))
 	default:
 		return
 	}