diff --git a/prospect_trace.py b/prospect_trace.py
index 866b171..b25a20a 100644
--- a/prospect_trace.py
+++ b/prospect_trace.py
@@ -41,7 +41,7 @@ class ProspectTrace(DeactivableMixin, ModelSQL, ModelView):
         super(ProspectTrace, cls).__setup__()
         cls._buttons.update({
             'wizard_schedule': {
-                'invisible': Eval('state') == 'open',
+                'invisible': Eval('state') == 'with_pending_calls',
                 }
         })
 
@@ -52,7 +52,7 @@ class ProspectTrace(DeactivableMixin, ModelSQL, ModelView):
     @classmethod
     @ModelView.button_action(
         'sale_opportunity_management.schedule_call_wizard')
-    def wizard_schedule():
+    def wizard_schedule(cls, prospect_traces):
         pass
 
     @fields.depends('calls', 'pending_call', 'current_interest', 'state')
diff --git a/view/prospect_trace_form.xml b/view/prospect_trace_form.xml
index 958fb07..8d083f9 100644
--- a/view/prospect_trace_form.xml
+++ b/view/prospect_trace_form.xml
@@ -25,6 +25,9 @@ this repository contains the full copyright notices and license terms. -->
         </page>
     </notebook>
 
+    <group col="-1" colspan="6" id="buttons">
+	    <button name="wizard_schedule" icon="tryton-calendar"/>
+    </group> 
     <label name="pending_call"/>
     <field name="pending_call"/>
     <label name="state"/>
diff --git a/view/schedule_start_form.xml b/view/schedule_start_form.xml
index e7422e2..43e69fa 100644
--- a/view/schedule_start_form.xml
+++ b/view/schedule_start_form.xml
@@ -4,5 +4,4 @@ this repository contains the full copyright notices and license terms. -->
 <form>
     <label name="date_time"/>
     <field name="date_time"/>
-    <button name="wizard_schedule" icon="tryton-forward"/>
 </form>
\ No newline at end of file