Fix add assets after refactor