Yes also think that's a good direction. That way, it would only really penalizes players who want the gold medal.
And players who only want to finish the level won't be forced to start over if they've made a mistake, which will be far less frustrating for them.