Change the key type from string to interface{}