Add conditional to load JS on module only if required