fix di container tests