diff --git a/internal/common/omci/image_test.go b/internal/common/omci/image_test.go
index b099675..d07ad51 100644
--- a/internal/common/omci/image_test.go
+++ b/internal/common/omci/image_test.go
@@ -17,11 +17,12 @@
 package omci
 
 import (
+	"testing"
+
 	"github.com/google/gopacket"
 	"github.com/opencord/omci-lib-go/v2"
 	me "github.com/opencord/omci-lib-go/v2/generated"
 	"gotest.tools/assert"
-	"testing"
 )
 
 func omciToStartSoftwareDownloadResponse(t *testing.T, omciPkt *gopacket.Packet) *omci.StartSoftwareDownloadResponse {
@@ -89,5 +90,5 @@
 	pkt, _, _ := ParseOpenOltOmciPacket(omciReqPkt)
 
 	count := ComputeDownloadSectionsCount(pkt)
-	assert.Equal(t, count, 1058)
+	assert.Equal(t, count, uint32(1058))
 }