Update for clara-storm API changes.