Fix parameter removal issue