Fix some module refereces