Lyon expecting Stokes to return in Ashes