Update test_package.py