Updated tests for docker