3 files changed, 9 insertions, 3 deletions
diff --git a/Cargo.toml b/Cargo.toml
index a7bbeab..eec44a8 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -22,6 +22,10 @@ two-face = { version = "0.4", default-features = false, features = [
 git = "https://github.com/pml68/iced"
 branch = "feat/rehighlight-on-redraw"
 
+[dev-dependencies.iced]
+git = "https://github.com/pml68/iced"
+branch = "feat/rehighlight-on-redraw"
+
 [package.metadata.docs.rs]
 rustdoc-args = ["--cfg", "docsrs"]
 all-features = true
diff --git a/README.md b/README.md
index 577e601..2ca543c 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,7 @@
 # iced_custom_highlighter
 
+[![builds.sr.ht status](https://builds.sr.ht/~pml68/iced_custom_highlighter.svg)](https://builds.sr.ht/~pml68/iced_custom_highlighter)
+
 A custom syntax highlighter for iced.
 
 It uses the colors from your app's Theme, based on a styling method (like `default_style`)
diff --git a/src/lib.rs b/src/lib.rs
index 94ff5e9..441df27 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -12,7 +12,7 @@
 //! #[derive(Default)]
 //! struct State {
 //!     content: text_editor::Content,
-//!     theme: Theme,
+//!     theme: Option<Theme>,
 //! }
 //!
 //! #[derive(Debug, Clone)]
@@ -34,7 +34,7 @@
 //!     )
 //!     .push(pick_list(
 //!         Theme::ALL,
-//!         Some(state.theme),
+//!         state.theme.clone(),
 //!         Message::ChangeTheme,
 //!     ))
 //!     .into()
@@ -47,7 +47,7 @@
 //!         }
 //!
 //!         Message::ChangeTheme(theme) => {
-//!             state.theme = theme;
+//!             state.theme = Some(theme);
 //!         }
 //!     }
 //! }