adding small fix to be able to run the umami network in DL2
tagging @fdibello @dguest @duperrin @sanmay