1 files changed, 2 insertions, 2 deletions
diff --git a/src/tool/mod.rs b/src/tool/mod.rs
index 4130244..b3fae80 100644
--- a/src/tool/mod.rs
+++ b/src/tool/mod.rs
@@ -20,7 +20,7 @@ pub use rect_room_tool::RectRoomTool;
 pub use selection_tool::SelectionTool;
 pub use wall_tool::WallTool;
 
-use crate::button::Button;
+use crate::input::Input;
 use crate::map::Map;
 use crate::math::Vec2;
 use crate::transform::Transform;
@@ -82,5 +82,5 @@ pub trait Tool {
 
     /// If there are any additional keybindings that need to be handled by this tool, these can be
     /// handled here.
-    fn on_button_pressed(&mut self, _map: &mut Map, _button: Button) {}
+    fn handle_custom_bindings(&mut self, _map: &mut Map, _input: &mut Input) {}
 }