diff --git a/rnnmorph/model.py b/rnnmorph/model.py
index 34b7d25..39bed98 100644
--- a/rnnmorph/model.py
+++ b/rnnmorph/model.py
@@ -378,7 +378,7 @@ def predict_probabilities(self, sentences: List[List[str]], batch_size: int,
 
         words = np.zeros((n_samples, max_sentence_len), dtype=int)
         grammemes = np.zeros((n_samples, max_sentence_len, self.grammeme_vectorizer_input.grammemes_count()),
-                             dtype=np.float)
+                             dtype=float)
         chars = np.zeros((n_samples, max_sentence_len, build_config.char_max_word_length), dtype=int)
 
         for i, sentence in enumerate(sentences):