diff --git a/report_close_statement.py b/report_close_statement.py
index bdece44..3c6d613 100644
--- a/report_close_statement.py
+++ b/report_close_statement.py
@@ -43,10 +43,11 @@ class ReportCloseStatementStart(ModelView):
     @classmethod
     def default_shop(cls):
         context = Transaction().context
-        shop = context['shop']
-
-        if shop:
-            return shop
+        if context:
+            shop = context['shop']
+            if shop:
+                return shop
+        return
 
 
 class PrintReportCloseStatement(Wizard):
@@ -108,10 +109,10 @@ class CashRegister(Report):
             return
         journals = [j.id for j in device.journals]
         statements = Statement.search([
-                    ('journal', 'in', journals),
-                    ('state', 'in', ['validated', 'posted']),
-                    ('date', '=', date)
-                ])
+            ('journal', 'in', journals),
+            ('state', 'in', ['validated', 'posted']),
+            ('date', '=', date)
+        ])
         return statements
 
     def _get_invoices_by_subtype(date, shop):
@@ -173,7 +174,7 @@ class CashRegister(Report):
             amount_tips.append(info_pizzas_sold['amount_tip'])
             total_amount += sale.total_amount
 
-        untaxed_amount_for_tax = round((tax_amount * 100)/Decimal('8.0'), 2)
+        untaxed_amount_for_tax = round((tax_amount * 100) / Decimal('8.0'), 2)
         statements = cls._get_statements_by_date(data['date'])
 
         if statements:
@@ -184,7 +185,7 @@ class CashRegister(Report):
                 'total_amount': sum(
                     (line.amount for line in statement.lines
                         if line.description), Decimal(0))
-                } for statement in statements]
+            } for statement in statements]
             report_context['total_statements'] = sum(
                 [s['total_amount'] for s in report_context['statements']])
 
diff --git a/sale.py b/sale.py
index e287d17..3e79ed3 100644
--- a/sale.py
+++ b/sale.py
@@ -126,10 +126,6 @@ class Sale(metaclass=PoolMeta):
 
     @classmethod
     def get_invoice(cls, record):
-        pool = Pool()
-        ctx = Transaction().context
-        Shop = pool.get('sale.shop')
-        shop = Shop.search([('id', '=', ctx["shop"])])[0]
         if record.state != 'draft' and record.invoices:
             invoice = record.invoices[0]
             data = {}
diff --git a/sale.xml b/sale.xml
index 5c88593..4700891 100644
--- a/sale.xml
+++ b/sale.xml
@@ -2,55 +2,57 @@
 <!--This file file is part of Tryton. The COPYRIGHT file at the top level
     of this repository contains the full copyright notices and license terms. -->
 <tryton>
-    <record model="ir.ui.view" id="sale_view_tree">
-        <field name="model">sale.sale</field>
-	<field name="inherit" ref="sale.sale_view_tree"/>
-        <field name="name">sale_tree</field>
-    </record>
-    <record model="ir.ui.view" id="sale_view_form">
-	<field name="model">sale.sale</field>
-	<field name="inherit" ref="sale.sale_view_form"/>
-	<field name="name">sale_form</field>
-    </record>
-    <record model="ir.ui.view" id="sale_line_view_form">
-	<field name="model">sale.line</field>
-	<field name="inherit" ref="sale.sale_line_view_form"/>
-	<field name="name">sale_line_form</field>
-    </record>
-    <record model="ir.model.button" id="sale_add_pizza_button">
-        <field name="name">add_pizza</field>
-        <field name="string">Add Pizza</field>
-        <field name="model" search="[('model', '=', 'sale.sale')]"/>
-    </record>
-    <record model="ir.model.button" id="sale_order_kitchen_button">
-        <field name="name">kitchen</field>
-        <field name="string">Kitchen</field>
-        <field name="model" search="[('model', '=', 'sale.sale')]"/>
-    </record>
-    <record model="ir.model.button" id="sale_order_bar_button">
-        <field name="name">bar</field>
-        <field name="string">Bar</field>
-        <field name="model" search="[('model', '=', 'sale.sale')]"/>
-    </record>
-    <record model="ir.model.button" id="sale_print_bill_button">
-        <field name="name">print_bill</field>
-        <field name="string">Bill</field>
-        <field name="model" search="[('model', '=', 'sale.sale')]"/>
-    </record>
-    <record model="ir.model.button" id="sale_impreso_button">
-        <field name="name">impreso</field>
-        <field name="string">Impreso</field>
-        <field name="model" search="[('model', '=', 'sale.sale')]"/>
-    </record>
-    <record model="ir.action.report" id="report_customer_order">
-        <field name="name">Customer Order</field>
-        <field name="model">sale.sale</field>
-        <field name="report_name">sale.customer_order</field>
-        <field name="report">sale_fast_food/report/customer_order.fodt</field>
-    </record>
-    <record model="ir.action.keyword" id="report_customer_order_keyword">
-        <field name="keyword">form_print</field>
-        <field name="model">sale.sale,-1</field>
-        <field name="action" ref="report_customer_order"/>
-    </record>
+    <data>
+	<record model="ir.ui.view" id="sale_view_tree">
+            <field name="model">sale.sale</field>
+	    <field name="inherit" ref="sale.sale_view_tree"/>
+            <field name="name">sale_tree</field>
+	</record>
+	<record model="ir.ui.view" id="sale_view_form">
+	    <field name="model">sale.sale</field>
+	    <field name="inherit" ref="sale.sale_view_form"/>
+	    <field name="name">sale_form</field>
+	</record>
+	<record model="ir.ui.view" id="sale_line_view_form">
+	    <field name="model">sale.line</field>
+	    <field name="inherit" ref="sale.sale_line_view_form"/>
+	    <field name="name">sale_line_form</field>
+	</record>
+	<record model="ir.model.button" id="sale_add_pizza_button">
+            <field name="name">add_pizza</field>
+            <field name="string">Add Pizza</field>
+            <field name="model" search="[('model', '=', 'sale.sale')]"/>
+	</record>
+	<record model="ir.model.button" id="sale_order_kitchen_button">
+            <field name="name">kitchen</field>
+            <field name="string">Kitchen</field>
+            <field name="model" search="[('model', '=', 'sale.sale')]"/>
+	</record>
+	<record model="ir.model.button" id="sale_order_bar_button">
+            <field name="name">bar</field>
+            <field name="string">Bar</field>
+            <field name="model" search="[('model', '=', 'sale.sale')]"/>
+	</record>
+	<record model="ir.model.button" id="sale_print_bill_button">
+            <field name="name">print_bill</field>
+            <field name="string">Bill</field>
+            <field name="model" search="[('model', '=', 'sale.sale')]"/>
+	</record>
+	<record model="ir.model.button" id="sale_impreso_button">
+            <field name="name">impreso</field>
+            <field name="string">Impreso</field>
+            <field name="model" search="[('model', '=', 'sale.sale')]"/>
+	</record>
+	<record model="ir.action.report" id="report_customer_order">
+            <field name="name">Customer Order</field>
+            <field name="model">sale.sale</field>
+            <field name="report_name">sale.customer_order</field>
+            <field name="report">sale_fast_food/report/customer_order.fodt</field>
+	</record>
+	<record model="ir.action.keyword" id="report_customer_order_keyword">
+            <field name="keyword">form_print</field>
+            <field name="model">sale.sale,-1</field>
+            <field name="action" ref="report_customer_order"/>
+	</record>
+    </data>
 </tryton>
diff --git a/view/sale_form.xml b/view/sale_form.xml
index 4604357..a161879 100644
--- a/view/sale_form.xml
+++ b/view/sale_form.xml
@@ -12,8 +12,9 @@ this repository contains the full copyright notices and license terms. -->
 	</group>
 	<newline/>
     </xpath>
-    <xpath expr="/form/notebook/page[@id='sale']/group[@id='amount']/field[@name='tax_amount']" position="after">
-		<label name="total_discount"/>
-		<field name="total_discount"/>
+    <xpath expr="/form/notebook/page[@id='sale']/group[@id='amount']/field[@name='tax_amount']"
+	   position="after">
+	<label name="total_discount"/>
+	<field name="total_discount"/>
     </xpath>
 </data>