improved assert_modules