Update for complete function bug