Main module fixes