diff --git a/api/vm.js b/api/vm.js
index f587e948..b5838473 100644
--- a/api/vm.js
+++ b/api/vm.js
@@ -176,7 +176,7 @@ ScratchTools.Scratch.waitForContextMenu = function (info) {
 };
 
 ScratchTools.Scratch.scratchPaint = function () {
-  var app = document.querySelector(".paint-editor_mode-selector_28iiQ");
+  var app = document.querySelector(".paint-editor_mode-selector_28iiQ")||document.querySelector(".paint-editor_mode-selector_O2uhP")||document.querySelector("[class*='paint-editor_mode-selector_']");
   if (app !== null) {
     return (
       app[