better fix for add button