Add missing package for test