Modify some function calls to match with new code.

To get worker from a message and to init messages returning to worker.

Signed-off-by: Huy Bui <huybui@anl.gov>