diff --git a/google/colab/_shell_customizations.py b/google/colab/_shell_customizations.py
index a5c0d46f..06eed5f6 100644
--- a/google/colab/_shell_customizations.py
+++ b/google/colab/_shell_customizations.py
@@ -152,7 +152,7 @@ def import_message(error):
                   'action_text': 'Install {}'.format(module_name),
                   # Snippets for installing a custom library always end with
                   # an import of the library itself.
-                  'snippet_filter': 'import {}'.format(module_name),
+                  'snippet_filter': module_name,
               },
           ],
       }