Fix test for module init