1 files changed, 5 insertions, 7 deletions

diff --git a/ext/js/pages/settings/mecab-controller.js b/ext/js/pages/settings/mecab-controller.js
index 4e2b02c6..9c55c9a0 100644
--- a/ext/js/pages/settings/mecab-controller.js
+++ b/ext/js/pages/settings/mecab-controller.js
@@ -16,23 +16,21 @@
  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
  */
 
+import {querySelectorNotNull} from '../../dom/query-selector.js';
 import {yomitan} from '../../yomitan.js';
 
 export class MecabController {
     constructor() {
-        /** @type {?HTMLButtonElement} */
-        this._testButton = null;
-        /** @type {?HTMLElement} */
-        this._resultsContainer = null;
+        /** @type {HTMLButtonElement} */
+        this._testButton = querySelectorNotNull(document, '#test-mecab-button');
+        /** @type {HTMLElement} */
+        this._resultsContainer = querySelectorNotNull(document, '#test-mecab-results');
         /** @type {boolean} */
         this._testActive = false;
     }
 
     /** */
     prepare() {
-        this._testButton = /** @type {HTMLButtonElement} */ (document.querySelector('#test-mecab-button'));
-        this._resultsContainer = /** @type {HTMLElement} */ (document.querySelector('#test-mecab-results'));
-
         this._testButton.addEventListener('click', this._onTestButtonClick.bind(this), false);
     }