diff --git a/README.md b/README.md
index 3e5b607..3c31e0e 100644
--- a/README.md
+++ b/README.md
@@ -72,17 +72,18 @@ Please cite the following papers if you are using the `dsm` package.
 
 [1] [Deep Survival Machines:
 Fully Parametric Survival Regression and
-Representation Learning for Censored Data with Competing Risks."
-arXiv preprint arXiv:2003.01176 (2020)](https://arxiv.org/abs/2003.01176)</a>
+Representation Learning for Censored Data with Competing Risks. (2021)](https://arxiv.org/abs/2003.01176)</a>
 
 ```
-  @article{nagpal2020deep,
-  title={Deep Survival Machines: Fully Parametric Survival Regression and\
-  Representation Learning for Censored Data with Competing Risks},
-  author={Nagpal, Chirag and Li, Xinyu and Dubrawski, Artur},
-  journal={arXiv preprint arXiv:2003.01176},
-  year={2020}
-  }
+@article{nagpal2021deep,
+  author={C. {Nagpal} and X. R. {Li} and A. {Dubrawski}},
+  journal={IEEE Journal of Biomedical and Health Informatics}, 
+  title={Deep Survival Machines: Fully Parametric Survival Regression and Representation Learning for Censored Data with Competing Risks}, 
+  year={2021},
+  volume={},
+  number={},
+  pages={1-1},
+  doi={10.1109/JBHI.2021.3052441}}
 ```
 
 Compatibility