fix bug with add button