Modify test-cases for the worker