Fix issue causing focus remaining on pen width button