Add missing import to test_skll