Fixing test case to get the container