A note on the extended Graetz problem