forgot to add new logic