Fix code example error with pickWhen