diff --git a/disentanglement_lib/evaluation/metrics/factor_vae.py b/disentanglement_lib/evaluation/metrics/factor_vae.py
index d920b4bf..3928eef4 100644
--- a/disentanglement_lib/evaluation/metrics/factor_vae.py
+++ b/disentanglement_lib/evaluation/metrics/factor_vae.py
@@ -127,7 +127,6 @@ def _compute_variances(ground_truth_data,
                                                 eval_batch_size)
   representations = np.transpose(representations)
   assert representations.shape[0] == batch_size
-  assert representations.shape[1] == ground_truth_data.num_factors
   return np.var(representations, axis=0, ddof=1)
 
 
diff --git a/setup.py b/setup.py
index 7acf9eb8..9be1ae7a 100644
--- a/setup.py
+++ b/setup.py
@@ -20,7 +20,7 @@
 
 setup(
     name='disentanglement_lib',
-    version='1.1',
+    version='1.2',
     description=('Library for research on disentangled representations.'),
     author='DisentanglementLib Authors',
     author_email='no-reply@google.com',