Module handling fixes