Daivari zadebiutował w dywizji cruiserweight brandu Raw 10 października podczas odcinka tygodniówki Raw, gdzie przegrał ze zwycięzcą Cruiserweight Classic TJ Perkinsem. Podczas pre-show gali Hell in a Cell on, Tony Nese i Drew Gulak przegrali z Cedrikiem Alexandrem, Lince'em Dorado i Sin Carą. Miesiąc później podczas pre-show gali Survivor Series w podobnym składzie przegrał z Perkinsem, Richem Swannem i Noamem Darem. Pod koniec roku rozpoczął rywalizację z Jackiem Gallagherem, którą przegrał z powodu porażki z Gallagherem w „I Forfeit” matchu.
W lipcu 2017 zaczął skupiać się na zostaniu pretendentem do WWE Cruiserweight Championship. 17 lipca zmierzył się z Akirą Tozawą, lecz pojedynek poddał menedżer Tozawy Titus O’Neil. Tydzień później miał odbyć się rewanż, lecz Daivari zaatakował Tozawę i posiadacza Cruiserweight Championship Neville’a. 25 lipca podczas odcinka tygodniówki 205 Live wygrał z Neville'em przez wyliczenie poza-ringowe. W listopadzie kiedy to Enzo Amore zdobył mistrzostwo cruiserweight, Daivari zaczął występować obok niego jako część grupy Zo Train.