Minor change to set boundary conditions