Fixes failing tests in test_studio_container.py