diff --git a/grader-notebook/Dockerfile b/grader-notebook/Dockerfile
index fbe535e..78895c5 100644
--- a/grader-notebook/Dockerfile
+++ b/grader-notebook/Dockerfile
@@ -9,8 +9,8 @@ ENV NB_GID=100
 
 # enable nbgrader's create assignment and formgrader extensions
 RUN jupyter nbextension enable --sys-prefix create_assignment/main \
- && jupyter serverextension enable --sys-prefix nbgrader.server_extensions.formgrader \
- && jupyter serverextension enable --sys-prefix nbgrader.server_extensions.course_list
+ && jupyter nbextension enable --sys-prefix formgrader/main --section=tree \
+ && jupyter serverextension enable --sys-prefix nbgrader.server_extensions.formgrader
 
 # disable the assignment extension, since graders don't need it
 RUN jupyter nbextension disable --sys-prefix assignment_list/main --section=tree