Add missing use statement in tests