add missing import to fix "-s" on regrade (fix #2303)

dumb little fix, @jrbl can you pull when you get a chance pls?