Corrected arguments for onMessage handler.