Adding import sys back

Oops, didn't notice that sys was removed from the latest commit.