diff --git a/pycave/bayes/gmm.py b/pycave/bayes/gmm.py
index 4c81da8..3b45f6b 100644
--- a/pycave/bayes/gmm.py
+++ b/pycave/bayes/gmm.py
@@ -3,9 +3,9 @@
 import torch
 import torch.nn as nn
 import torch.distributions as dist
-import bxtorch.nn as xnn
-from bxtorch.utils.stdio import ProgressBar
-from bxtorch.utils.torch import to_one_hot
+import pyblaze.nn as xnn
+from pyblaze.utils.stdio import ProgressBar
+from pyblaze.utils.torch import to_one_hot
 from sklearn.cluster import KMeans
 from .utils import log_normal, log_responsibilities, max_likeli_means, max_likeli_covars
 
@@ -166,7 +166,7 @@ def fit(self, data, max_iter=100, batch_size=None, eps=1e-7, verbose=False):
 
         Returns
         -------
-        bxtorch.nn.History
+        pyblaze.nn.History
             A history object containing the negative log-likelihoods over the course of the
             EM-training.
         """
diff --git a/pycave/bayes/markov.py b/pycave/bayes/markov.py
index 6c4b0d4..6e48ba4 100644
--- a/pycave/bayes/markov.py
+++ b/pycave/bayes/markov.py
@@ -2,7 +2,7 @@
 import torch
 import torch.nn as nn
 import torch.distributions as dist
-import bxtorch.nn as xnn
+import pyblaze.nn as xnn
 from .utils import power_iteration
 
 class MarkovModelConfig(xnn.Config):
@@ -77,7 +77,7 @@ def fit(self, sequences, teleport_alpha=0):
 
         Returns
         -------
-        bxtorch.nn.History
+        pyblaze.nn.History
             For completeness, it returns a history object. However, apart from the duration of the
             training, the object does not carry any information.
         """