Add more expectations