Updated to work with new Omnium logging