More module fixes