Makes edit button consistent w/ UI