Small refactor to assets task