It makes sense Yang Kai is still greatly weakened after the battle in the dream/illusion. It's also quite convenient for the story because otherwise, apart from the great emperors (and those of similar power, like divine spirits), nobody would be his opponent anymore.