Use update_attribute, missing line at file end