Make tests happy

Make tests happy