Adding missing mock