PA (state & local) Taxable Gross Definitions table when have 401K before & after tax deductions

We have Before and After Tax 401K deductions and both fall under the 401K deduction code. In Pennsylvania (state and locals) consider the Before (pre-tax) 401K to be taxable. Essentially in the Taxable Gross Definition Table, we can't figure out if or how we can define the criteria differently for the two different types of 401K. Essentially for the Before Tax, we want the state & local PA withholding to include this as taxable wages, but for the After Tax, we would not. Has anyone else encountered this issue?