Fix null reference when no expectations set