Fix classification button issue.