diff --git a/src/pages/[platform]/build-a-backend/add-aws-services/predictions/set-up-predictions/index.mdx b/src/pages/[platform]/build-a-backend/add-aws-services/predictions/set-up-predictions/index.mdx
index 9b7642739dd..dd8f357df21 100644
--- a/src/pages/[platform]/build-a-backend/add-aws-services/predictions/set-up-predictions/index.mdx
+++ b/src/pages/[platform]/build-a-backend/add-aws-services/predictions/set-up-predictions/index.mdx
@@ -145,7 +145,7 @@ backend.addOutput({
 To install the Amplify library to use predictions features, run the following commands in your project's root folder:
 
 ```bash title="Terminal" showLineNumbers={false}
-npm add aws-amplify
+npm add aws-amplify @aws-amplify/predictions
 ```
 
 ## Configure the frontend
@@ -153,13 +153,13 @@ npm add aws-amplify
 Import and load the configuration file in your app. It is recommended you add the Amplify configuration step to your app's root entry point. For example `main.ts` in React and Angular.
 
 ```ts title="src/main.ts"
-import { Predictions } from "aws-amplify/predictions";
+import { Amplify } from "aws-amplify";
 import outputs from "./amplify_outputs.json";
 
 Amplify.configure(outputs);
 
 Amplify.configure({
   ...Amplify.getConfig(),
-  Predictions: config.custom.Predictions,
+  Predictions: outputs.custom.Predictions,
 });
 ```