Fix test failing on Drone build
Fix test failing on Drone build