Fix compilation error due to changes in types.go

I'm fairly sure this fixes the issue, but hopefully you can do some more testing on your end.