TS: Fixed it so edit button actually changes when you're in editting mode