correctly handle set(key, value) instead of set({key: value})