diff --git a/src/document.jl b/src/document.jl
index 7c89e582..34ae170e 100644
--- a/src/document.jl
+++ b/src/document.jl
@@ -171,7 +171,8 @@ A NGramDocument{AbstractString}
  * Title: Untitled Document
  * Author: Unknown Author
  * Timestamp: Unknown Time
- * Snippet: ***SAMPLE TEXT NOT AVAILABLE***```
+ * Snippet: ***SAMPLE TEXT NOT AVAILABLE***
+```
 """
 function NGramDocument(txt::AbstractString, dm::DocumentMetadata, n::Integer...=1)
     NGramDocument(ngramize(dm.language, tokenize(dm.language, String(txt)), n...), (length(n) == 1) ? Int(first(n)) : Int[n...], dm)
diff --git a/test/crf.jl b/test/crf.jl
index 5c830d71..8225a8ad 100644
--- a/test/crf.jl
+++ b/test/crf.jl
@@ -1,5 +1,5 @@
 using Flux
-using Flux: gradient, LSTM, Dense, reset!
+using Flux: gradient, LSTM, Dense, reset!, onehot
 using TextAnalysis: score_sequence, forward_score
 
 @testset "crf" begin
@@ -148,4 +148,3 @@ using TextAnalysis: score_sequence, forward_score
         @test crf_param_1 != crf_param_2
     end
 end
-