diff --git a/segment_core_string.go b/segment_core_string.go
index 036e2bb..60742f4 100644
--- a/segment_core_string.go
+++ b/segment_core_string.go
@@ -38,7 +38,7 @@ type CoreString struct {
 
 type PubRestriction struct {
 	PurposeId       int
-	RestrictionType int
+	RestrictionType RestrictionType
 	NumEntries      int
 	RangeEntries    []*RangeEntry
 }