diff --git a/sandbox/react-flushSync/hooks/useAfterNextLayout.js b/sandbox/react-flushSync/hooks/useAfterNextLayout.js
index a788159..a314df7 100644
--- a/sandbox/react-flushSync/hooks/useAfterNextLayout.js
+++ b/sandbox/react-flushSync/hooks/useAfterNextLayout.js
@@ -1,6 +1,6 @@
-import { useLayoutEffect, useRef, useState } from "react";
+import { useLayoutEffect, useRef } from "react";
 
-export function useAfterNextLayoutWithRefs() {
+export function useAfterNextLayout() {
 	const callbacks = useRef([]);
 
 	useLayoutEffect(() => {
@@ -15,20 +15,4 @@ export function useAfterNextLayoutWithRefs() {
 	}
 }
 
-function useAfterNextLayoutWithState() {
-	const [callbacks, setCallbacks] = useState([]);
-
-	useLayoutEffect(() => {
-		if (callbacks.length === 0) return;
-		for (let callback of callbacks) {
-			callback();
-		}
-		setCallbacks([]);
-	}, [callbacks]);
-
-	return (callback) => {
-		setCallbacks((callbacks) => [...callbacks, callback]);
-	}
-}
-
-export { useAfterNextLayoutWithRefs as useAfterNextLayout };
\ No newline at end of file
+export default useAfterNextLayout;
\ No newline at end of file