fix check for add button