Gravar-mail: Accounting for focal shift in the Shack–Hartmann wavefront sensor