fixed another small bug with z2