[STUDIO-2005]Update tests to follow modification of UI