diff --git a/share/availability/light/availability.go b/share/availability/light/availability.go
index 6a23abe3d9..64e8248a06 100644
--- a/share/availability/light/availability.go
+++ b/share/availability/light/availability.go
@@ -103,7 +103,7 @@ func (la *ShareAvailability) SharesAvailable(ctx context.Context, header *header
 		}
 		// Verify total samples count.
 		totalSamples := len(samples.Remaining) + len(samples.Available)
-		if (totalSamples != int(la.params.SampleAmount)) && (totalSamples != len(dah.RowRoots)*len(dah.RowRoots)) {
+		if (totalSamples != int(la.params.SampleAmount)) && (la.params.SampleAmount > len(dah.RowRoots)*len(dah.RowRoots)) {
 			return fmt.Errorf("invalid sampling result:"+
 				" expected %d samples, got %d", la.params.SampleAmount, totalSamples)
 		}