diff --git a/src/rimestate.cpp b/src/rimestate.cpp
index 7ca75be..b2ab9cd 100644
--- a/src/rimestate.cpp
+++ b/src/rimestate.cpp
@@ -418,7 +418,7 @@ void RimeState::restore() {
     if (savedCurrentSchema_.empty()) {
         return;
     }
-    if (!engine_->schemas().count(savedCurrentSchema_)) {
+    if (!std::count(engine_->schemas().begin(), engine_->schemas().end(), savedCurrentSchema_) {
         return;
     }