small change to allow button on custom