Fix for assets registration to run on correct action