Fixed minor error with the edit buttons displaying all the time.