diff --git a/courses/react-js/advanced-level/component-optimization/MemoizationExample.js b/courses/react-js/advanced-level/component-optimization/MemoizationExample.js
index 4d0a596e1..e885d7748 100644
--- a/courses/react-js/advanced-level/component-optimization/MemoizationExample.js
+++ b/courses/react-js/advanced-level/component-optimization/MemoizationExample.js
@@ -1,10 +1,16 @@
 import React, { useMemo, useState } from "react";
+import PropTypes from 'prop-types';
+
 
 const ExpensiveComponent = ({ value }) => {
   const expensiveFunction = (value) => {
     // Expensive computation
     return value * 2;
   };
+  
+  ExpensiveComponent.propTypes = {
+    value: PropTypes.number.isRequired
+  };
   const memoizedValue = useMemo(() => expensiveFunction(value), [value]);
   return <div>{memoizedValue}</div>;
 };