[Studio] Fix some tests