1 files changed, 4 insertions, 2 deletions
diff --git a/gtk2_ardour/time_info_box.h b/gtk2_ardour/time_info_box.h
index 7409d7631e..035cdb7e00 100644
--- a/gtk2_ardour/time_info_box.h
+++ b/gtk2_ardour/time_info_box.h
@@ -29,6 +29,8 @@
 #include "ardour/ardour.h"
 #include "ardour/session_handle.h"
 
+#include "ardour_button.h"
+
 namespace ARDOUR {
 	class Session;
         class Location;
@@ -69,8 +71,8 @@ class TimeInfoBox : public Gtk::HBox, public ARDOUR::SessionHandlePtr
     PBD::ScopedConnectionList punch_connections;
     PBD::ScopedConnectionList editor_connections;
 
-    Gtkmm2ext::StatefulToggleButton punch_in_button;
-    Gtkmm2ext::StatefulToggleButton punch_out_button;
+    ArdourButton punch_in_button;
+    ArdourButton punch_out_button;
 
     void selection_changed ();