comment out duplicate (bypass) put_z stage for now