diff --git a/test/highlight_test.rb b/test/highlight_test.rb
index ae8e404..f515f6a 100644
--- a/test/highlight_test.rb
+++ b/test/highlight_test.rb
@@ -50,6 +50,7 @@ class HighlightTest < Minitest::Test
   def test_multiple_words
     store_names ["Hello World Hello"]
     assert_equal "<em>Hello</em> World <em>Hello</em>", Product.search("hello", highlight: true).highlights.first[:name]
+    assert_equal "<em>Hello</em> World <em>Hello</em>", Product.search("hello", relation: true).highlight(true).highlights.first[:name]
   end
 
   def test_encoder
@@ -60,6 +61,7 @@ class HighlightTest < Minitest::Test
   def test_word_middle
     store_names ["Two Door Cinema Club"]
     assert_equal "Two Door <em>Cinema</em> Club", Product.search("ine", match: :word_middle, highlight: true).highlights.first[:name]
+    assert_equal "Two Door <em>Cinema</em> Club", Product.search("ine", relation: true).match(:word_middle).highlight(true).highlights.first[:name]
   end
 
   def test_body
--
libgit2 0.21.0