updated module so would work