Fix regression that @geetarista fixed.