diff --git a/app/assets/javascripts/discourse/widgets/post-small-action.js.es6 b/app/assets/javascripts/discourse/widgets/post-small-action.js.es6
index a09f6df7f90..2f26fac0986 100644
--- a/app/assets/javascripts/discourse/widgets/post-small-action.js.es6
+++ b/app/assets/javascripts/discourse/widgets/post-small-action.js.es6
@@ -75,6 +75,7 @@ export default createWidget('post-small-action', {
 
     if (attrs.canDelete) {
       contents.push(this.attach('button', {
+        className: 'small-action-delete',
         icon: 'times',
         action: 'deletePost',
         title: 'post.controls.delete'
@@ -83,6 +84,7 @@ export default createWidget('post-small-action', {
 
     if (attrs.canEdit) {
       contents.push(this.attach('button', {
+        className: 'small-action-edit',
         icon: 'pencil',
         action: 'editPost',
         title: 'post.controls.edit'