Add missing `sys` import