Bug fix: import sys module