Fix scale to use vec instead of float