Update to work with recent version of Stem