Updated the import in remove_caps_test to import two different types of remove_caps operations