Add conditional to fix multiple hosts