diff --git a/services/comfy/entrypoint.sh b/services/comfy/entrypoint.sh
index a962ac5..8161566 100755
--- a/services/comfy/entrypoint.sh
+++ b/services/comfy/entrypoint.sh
@@ -47,8 +47,8 @@ done
 if [ "$(ls -A /stable-diffusion/custom_nodes)" ]; then
   chmod 777 -R "/stable-diffusion/custom_nodes/"
   apt-get install build-essential -y
-  for dir in "/stable-diffusion/custom_nodes/*/"; do
-    if [ -d $dir ]; then
+  for dir in "/stable-diffusion/custom_nodes/*"; do
+    if [ -e "$dir/requirements.txt" ]; then
       echo $dir
       cd $dir
       pip install -r requirements.txt