Bugfix: forgot to import sys