Should fix output issues for UpdateModule