diff --git a/documentation.md b/documentation.md
index daac9c5d..3cde44a6 100644
--- a/documentation.md
+++ b/documentation.md
@@ -22,7 +22,7 @@ const bodySegmentation = ml5.bodySegmentation(?modelName, ?options, ?callback);
 
 **Parameters:**
 
-- **modelName**: OPTIONAL: A string specifying which model to use, "SelfieSegmentation" or "BodyPix".
+- **modelName**: OPTIONAL: A string specifying which model to use, "SelfieSegmentation" or "BodySegmentation".
 
 - **options**: OPTIONAL. An object to change the default configuration of the model. See the example options:
 
diff --git a/src/index.js b/src/index.js
index ec993115..2f896780 100644
--- a/src/index.js
+++ b/src/index.js
@@ -7,7 +7,7 @@ import * as tf from "@tensorflow/tfjs";
 import * as tfvis from "@tensorflow/tfjs-vis";
 import p5Utils from "./utils/p5Utils";
 import setBackend from "./utils/setBackend";
-import bodyPix from "./BodySegmentation";
+import bodySegmentation from "./BodySegmentation";
 import communityStatement from "./utils/communityStatement";
 import imageClassifier from "./ImageClassifier";
 import preloadRegister from "./utils/p5PreloadHelper";
@@ -25,7 +25,7 @@ export default Object.assign({ p5Utils }, preloadRegister(withPreload), {
   faceMesh,
   bodyPose,
   setBackend,
-  bodyPix,
+  bodySegmentation,
 });
 
 communityStatement();