Update to work with new GershwinStack implementation