prompt pruning fixes