Fixed a bug, didn't import sys properly