Fix base module not being included in setup