diff --git a/client/modules/IDE/components/Header/Toolbar.jsx b/client/modules/IDE/components/Header/Toolbar.jsx
index 16ed74ff3e..ce72d67a35 100644
--- a/client/modules/IDE/components/Header/Toolbar.jsx
+++ b/client/modules/IDE/components/Header/Toolbar.jsx
@@ -73,6 +73,7 @@ const Toolbar = (props) => {
       <button
         className={playButtonClass}
         id="play-sketch"
+        data-testid="play-button"
         onClick={() => {
           props.syncFileContent();
           dispatch(startSketch());
@@ -85,6 +86,7 @@ const Toolbar = (props) => {
       </button>
       <button
         className={stopButtonClass}
+        data-testid="stop-button"
         onClick={() => dispatch(stopSketch())}
         aria-label={t('Toolbar.StopSketchARIA')}
         title={t('Toolbar.StopSketchARIA')}