Add missing require to fix test :bomb: