Update module to reflect new case API