Updated to handle change in diary object class name