diff --git a/README.md b/README.md
index 9890b2d..c6997a8 100644
--- a/README.md
+++ b/README.md
@@ -109,9 +109,9 @@ y_val = [[0.5], [1.5]]
 N_trial = 10
 
 grea_model.autofit(
-    X_train=X_train.tolist(),
+    X_train=X_train,
     y_train=y_train,
-    X_val=X_val.tolist(),
+    X_val=X_val,
     y_val=y_val,
     n_trials=N_trial,
 )
@@ -223,4 +223,4 @@ See the structure of `torch_molecule` with the command `tree -L 2 torch_molecule
 
 ## Acknowledgements
 
-The project template was adapted from [https://github.com/lwaekfjlk/python-project-template](https://github.com/lwaekfjlk/python-project-template). We thank the authors for their contribution to the open-source community.
\ No newline at end of file
+The project template was adapted from [https://github.com/lwaekfjlk/python-project-template](https://github.com/lwaekfjlk/python-project-template). We thank the authors for their contribution to the open-source community.