Remove .is_infinitesimal assumption