Fixing controllers behavior when update.