In the worst case with 170 balls, your solution actually needs 6 uses of the scale.

Hint: see what happens in your solution if the last ball is heaviest