Updates for new z probe