Fix tests by adding missing package