fix bugs of with expression