Module layout fixes