diff --git a/README.md b/README.md
index 93f8c1cee..6f49d3820 100644
--- a/README.md
+++ b/README.md
@@ -101,7 +101,7 @@ If you find our work useful, please consider citing us!
 
 ```bibtex
 @article{toshniwal2024openmathinstruct2,
-  title   = {OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Data},
+  title   = {{OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Data}},
   author  = {Shubham Toshniwal and Wei Du and Ivan Moshkov and Branislav Kisacanin and Alexan Ayrapetyan and Igor Gitman},
   year    = {2024},
   journal = {arXiv preprint arXiv: Arxiv-2410.01560}
@@ -109,11 +109,11 @@ If you find our work useful, please consider citing us!
 ```
 
 ```bibtex
-@article{toshniwal2024openmathinstruct1,
-  title   = {OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Dataset},
+@inproceedings{toshniwal2024openmathinstruct1,
+  title   = {{OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Dataset}},
   author  = {Shubham Toshniwal and Ivan Moshkov and Sean Narenthiran and Daria Gitman and Fei Jia and Igor Gitman},
   year    = {2024},
-  journal = {arXiv preprint arXiv: Arxiv-2402.10176}
+  booktitle = {Advances in Neural Information Processing Systems},
 }
 ```