Quick type fix

This is a quick type fix that i missed when i did the ImmutableMap integration.