Fix module inclusion