Small update for stability