Improve UDMB initialization