adjust autoreload to work in html mode

1 files changed, 2 insertions, 2 deletions

diff --git a/cgi/templates/main.html.ep b/cgi/templates/main.html.ep
index 821824a..7a2612d 100644
--- a/cgi/templates/main.html.ep
+++ b/cgi/templates/main.html.ep
@@ -66,7 +66,7 @@
 id="display" height="<%= $height * 4 %>" width="<%= $width * 4 %>"/>
 % }
 % elsif ($frontend eq 'html') {
-<iframe src="/<%= $city %>/<%= $stop %>.html?<%= $params %>"
+<iframe src="/<%= $city %>/<%= $stop %>.html?<%= $params %>" id="display"
 height="<%= $height * 5 %>" width="<%= $width * 5 %>"></iframe>
 % }
 % }
@@ -157,7 +157,7 @@ v<%= $version %>
 <script type="text/javascript">
 function reloadDisplay() {
 	d = new Date();
-	$("#display").attr("src", "../<%= $city %>/<%= $stop %>.png?<%= $params %>&r="+d.getTime())
+	$("#display").attr("src", "../<%= $city %>/<%= $stop %>.<%= $frontend %>?<%= $params %>&r="+d.getTime())
 	setTimeout('reloadDisplay()', 60000);
 }