diff --git a/commerce_line_item_type.features.inc b/commerce_line_item_type.features.inc
index 13bfd2a..98413d4 100644
--- a/commerce_line_item_type.features.inc
+++ b/commerce_line_item_type.features.inc
@@ -54,7 +54,7 @@ function commerce_line_item_type_features_export_render($module, $data) {
   // Filter out all line item types not coming from UI.
   foreach ($data as $type) {
    if (isset($info[$type]) && $line_item_type = $info[$type]) {
-     if (isset($line_item_type['module']) && $line_item_type['module'] == 'commerce_custom_line_items') {
+     if (isset($line_item_type['module']) && ($line_item_type['module'] == 'commerce_custom_line_items' || $line_item_type['module'] == 'commerce_custom_product')) {
       $exportable_data[$type] = $type;
      }
    }
@@ -91,7 +91,11 @@ function commerce_line_item_type_features_revert($module) {
         $type['is_new'] = TRUE;
       }
       // Use UI function because it provides already the logic we need
-      module_invoke('commerce_custom_line_items', 'commerce_line_item_type_save', $type);
+      if (module_exists('commerce_custom_product')) {
+        module_invoke('commerce_custom_product', 'line_item_type_save', $type);
+      } elseif (module_exists('commerce_line_item')) {
+        module_invoke('commerce_custom_line_items', 'commerce_line_item_type_save', $type);
+      }
     }
   }
   else {