Updated the setCallback function to use the renamed __callback