diff --git a/actions/uv/jupyter/action.yml b/actions/uv/jupyter/action.yml
index ac7d473..d49242c 100644
--- a/actions/uv/jupyter/action.yml
+++ b/actions/uv/jupyter/action.yml
@@ -15,15 +15,6 @@ name: Build the Jupyter Book
 
 description: "Build jupyter book"
 
-on:
-  workflow_call:
-    inputs:
-      kernel:
-        description: 'Jupyter kernel'
-        type: string
-        required: false
-        default: 'cvx'
-
 runs:
     using: "composite"
 
@@ -37,8 +28,6 @@ runs:
         shell: bash
         run: |
           uv pip install jupyter-book --system
-          # There is no need to create the kernel if the jupyter book is installed within the same virtual environment as the package
-          # poetry run python -m ipykernel install --user --name=${{ inputs.kernel || 'cvx' }}
 
       # Build the book
       - name: Build the book
diff --git a/actions/uv/marimo/action.yml b/actions/uv/marimo/action.yml
index ab31889..99a230e 100644
--- a/actions/uv/marimo/action.yml
+++ b/actions/uv/marimo/action.yml
@@ -18,6 +18,12 @@ description: "Convert files for Marimo"
 on:
   workflow_call:
     inputs:
+      marimo-folder:
+        description: 'Folder with *.py files for Marimo'
+        type: string
+        required: false
+        default: 'book/marimo'
+
       python-version:
         description: 'Python version'
         type: string
@@ -43,7 +49,7 @@ runs:
           mkdir -p artifacts/marimo
 
           # Loop through all .py files
-          for file in book/marimo/*.py; do
+          for file in ${{ marimo-folder || 'book/marimo' }}/*.py; do
 
               echo $file