Fix compilation error due to changes in types.go