Fix the vacuum oscillation example