Fix add button problem