diff --git a/website/docs/API.md b/website/docs/API.md
index 071095df8..0199a076d 100644
--- a/website/docs/API.md
+++ b/website/docs/API.md
@@ -294,12 +294,13 @@ import { FlatList, View } from 'react-native';
 import { render, fireEvent } from '@testing-library/react-native';
 
 const onEndReached = jest.fn();
-const { getByType } = render(
+const { getByTestId } = render(
   <FlatList
     data={Array.from({ length: 10 }, (_, key) => ({ key: `${key}` }))}
     renderItem={() => <View style={{ height: 500, width: 100 }} />}
     onEndReached={onEndReached}
     onEndReachedThreshold={0.2}
+    testID="flat-list"
   />
 );
 const eventData = {
@@ -320,7 +321,7 @@ const eventData = {
   },
 };
 
-fireEvent.scroll(getByType(ScrollView), eventData);
+fireEvent.scroll(getByTestId('flat-list'), eventData);
 expect(onEndReached).toHaveBeenCalled();
 ```