diff --git a/system/admin/views/edit-page.html.php b/system/admin/views/edit-page.html.php
index e0f0e26..46e1415 100644
--- a/system/admin/views/edit-page.html.php
+++ b/system/admin/views/edit-page.html.php
@@ -118,12 +118,13 @@ $images = get_gallery();
                         <input type="submit" name="submit" class="btn btn-primary submit" value="<?php echo i18n('Save_category');?>"/>
                     <?php } else {?>
                         <input type="hidden" name="oldfile" class="text" value="<?php echo $url ?>"/>
-                        <?php $count = count(find_subpage($oldmd)); ?>
+                        <?php $dd = find_subpage($oldmd); ?>
+                        <?php $dr = find_draft_subpage($oldmd);?>
                         <?php if (stripos($dir . '/', '/draft/') !== false) { ?>
-                        <input type="submit" name="publishdraft" class="btn btn-primary submit" value="<?php echo i18n('Publish_draft');?>"/> <input type="submit" name="updatedraft" class="btn btn-primary draft" value="<?php echo i18n('Update_draft');?>"/> <?php if ($count == 0 && $type != 'is_page'):?><a class="btn btn-danger" href="<?php echo $delete ?>"><?php echo i18n('Delete');?></a><?php endif;?>        
-                        
+                        <input type="submit" name="publishdraft" class="btn btn-primary submit" value="<?php echo i18n('Publish_draft');?>"/> <input type="submit" name="updatedraft" class="btn btn-primary draft" value="<?php echo i18n('Update_draft');?>"/> <?php if (empty($dd) && empty($dr) && $type != 'is_page'):?><a class="btn btn-danger" href="<?php echo $delete ?>"><?php echo i18n('Delete');?></a><?php endif;?>
+
                         <?php } else { ?>
-                        <input type="submit" name="submit" class="btn btn-primary submit" value="<?php echo i18n('Save');?>"/> <input type="submit" name="revertpage" class="btn btn-primary revert" value="<?php echo i18n('Revert_to_draft');?>"/> <?php if ($count == 0 && $type != 'is_page'):?><a class="btn btn-danger" href="<?php echo $delete ?>"><?php echo i18n('Delete');?></a><?php endif;?>
+                        <input type="submit" name="submit" class="btn btn-primary submit" value="<?php echo i18n('Save');?>"/> <?php if (empty($dd) && empty($dr) && $type != 'is_page'):?><input type="submit" name="revertpage" class="btn btn-primary revert" value="<?php echo i18n('Revert_to_draft');?>"/> <a class="btn btn-danger" href="<?php echo $delete ?>"><?php echo i18n('Delete');?></a><?php endif;?>
                         <?php } ?>
                     <?php } ?>
                 </div>
@@ -207,4 +208,4 @@ $images = get_gallery();
   $('.the-img').click(function(e) {
     $('#insertImageDialogURL').val($(e.target).attr('src'));
   });
-</script>
\ No newline at end of file
+</script>