Bug 1233878 - tracking flags don't show up in the view of the bug right after filing

diff --git a/extensions/BugModal/Extension.pm b/extensions/BugModal/Extension.pm
index e929f3fcbb5f4f02e812d0936c5ad0e577a2ded1..b1356bc34a96d59f5c28d120180f53c37d6305ff 100644 (file)
--- a/extensions/BugModal/Extension.pm
+++ b/extensions/BugModal/Extension.pm
@@ -165,9 +165,17 @@ sub template_before_process {
         return;
     }
 
-    return unless $file =~ m#^bug/show-([^\.]+)\.html\.tmpl$#;
-    my $format = $1;
-    return unless _alternative_show_bug_format() eq $format;
+    if ($file =~ m#^bug/show-([^\.]+)\.html\.tmpl$#) {
+        my $format = $1;
+        return unless _alternative_show_bug_format() eq $format;
+    }
+    elsif ($file ne 'bug_modal/edit.html.tmpl') {
+        return;
+    }
+
+    if ($vars->{bug} && !$vars->{bugs}) {
+        $vars->{bugs} = [$vars->{bug}];
+    }
 
     return unless
         $vars->{bugs}