Use proper types for callbacks in demo