--- uc_aac.js	2010-02-22 20:50:44.000000000 +0100
+++ uc_aac.corrected.js	2010-02-22 20:52:12.000000000 +0100
@@ -16,7 +16,11 @@ function uc_aac_calculate(element) {
       // Update the add to cart form.
       if (data.form) {
         var action = form.attr('action');
-        $(form).after(data.form).next().attr('action', action).uc_aac_attach();
+        if (UCOI) {
+        	$(form).after(data.form).next().attr('action', action).uc_aac_attach().uc_ucoi_attach();
+        } else {
+        	$(form).after(data.form).next().attr('action', action).uc_aac_attach();
+        }
         form.remove();
       }
     }
@@ -30,6 +34,7 @@ jQuery.fn.uc_aac_attach = function() {
   $(this).find('input:radio[@name^=attributes], input:checkbox[@name^=attributes]').click(function() {
     uc_aac_calculate(this);
   });
+  return this;
 }
 
 $(function() {