Make max_pool stable