Setting up a water aspirator pump for vacuum distillation
Since you want the distillate, use the following setup. 



This way if the water pressure drops then the water is sucked into the trap flask, not the distillate.