Fix using .set(obj, cb) (fixes #3)