Add conditional in modules