Fix bug in .set()