Update logic again