Corrected remove method to retrieve actual classes from this API.