Fix removeButtonActionListener