add watch section and bench section

1 files changed, 3 insertions, 3 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 16b1d25..7dd08e9 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -25,17 +25,17 @@ fn document() -> web_sys::Document {
 }
 
 fn canvas() -> JSResult<web_sys::HtmlCanvasElement> {
-    let x = document().query_selector("canvas")?.ok_or("no canvas")?;
+    let x = document().query_selector("section.watch canvas")?.ok_or("no canvas")?;
     Ok(x.dyn_into::<web_sys::HtmlCanvasElement>()?)
 }
 
 fn fps() -> JSResult<web_sys::HtmlElement> {
-    let x = document().query_selector("#fps")?.ok_or("no fps counter")?;
+    let x = document().query_selector("section.watch .fps")?.ok_or("no fps counter")?;
     Ok(x.dyn_into::<web_sys::HtmlElement>()?)
 }
 
 fn go() -> JSResult<web_sys::HtmlElement> {
-    let x = document().query_selector("button")?.ok_or("no go button")?;
+    let x = document().query_selector("section.watch button")?.ok_or("no go button")?;
     Ok(x.dyn_into::<web_sys::HtmlElement>()?)
 }