Update to bug fix