Abstract We revisit the classical $K_Z$ problem — determination of the vertical force and implied total mass density distribution of the Milky Way disk — for a wide range of Galactocentric radius and vertical height using chemically selected thin and thick disk samples based on APOGEE spectroscopy combined with the Gaia astrometry. We derived the velocity dispersion profiles in Galactic cylindrical coordinates, and solved the Jeans Equation for the two samples separately.