Update for new codebox version