V. Kaloshin (2005) gave a method to Whitney stratify semi-algebraic sets. We correct his proof and give a more general result applying to definable sets in arbitrary o-minimal structures. (Work in collaboration with Nguyen Xuan Viet Nhan and Saurabh Trivedi.)