Replace check key and set with .get()