diff --git a/python/sandbox/cos_sandbox.py b/python/sandbox/cos_sandbox.py
index a25035544..90259b353 100644
--- a/python/sandbox/cos_sandbox.py
+++ b/python/sandbox/cos_sandbox.py
@@ -73,7 +73,9 @@ def setJS(jsStr):
 													'functools', 'itertools', 'operator', 'string',
 													'collections', 're', 'json', 'csv','copy','copyreg', 'autograde_runes',
 													'engine','hungry_games_classes','hungry_games','simulation',
-													'heapq', 'bisect','inspect','__future__', 'generic_arith_min', 'encodings.idna', 'encodings', 'urllib', 'urllib.error', 'buses', 'uuid')
+													'heapq', 'bisect','inspect','__future__', 'generic_arith_min',
+													'encodings.idna', 'encodings', 'urllib', 'urllib.error', 'buses', 'uuid',
+													'autograde_hi_graph')
 
 # whitelist of custom modules to import into OPT
 # (TODO: support modules in a subdirectory, but there are various