Probably shouldn't be using a button on edit mode anyways...