Just a minor struct change

Removed a map since we are using a different one, seems fair.