Most of the Proboscis monkey (Nasalis larvatus) habitat is outside the protected area, so the potential for destruction is high. One of the efforts to improve its habitat is habitat restoration. This study aims to determine plant species for proboscis monkey habitat restoration based on reference habitats. The research was conducted at eight locations in East Kalimantan, Indonesia i.e. Kuala Samboja, Mahakam Delta, Suwi River, Ohong River, Berau Delta (mangrove and riverine), Sungai Wain, and Kutai National Park. Vegetation analysis with line-plot sampling method with a total sample of all locations of 11.7 ha. The results showed that the habitat included three main types: mangrove, riverine, and riverine-near coastal. A total of 152 tree species, 41 of which were potential proboscis monkey feed. The highest plant species diversity value was in the riverine forest-Berau Delta of 1.25, while the lowest was in the mangrove forest-Berau Delta of 0.54. Species that are a food source and have good adaptation in riverine habitat are Vitex pinnata, Dillenia excelsa, Garcinia spp, Heritiera littoralis, Ilex cymosa, Cerbera manghas, Hevea braziliensis, and Sonneratia caseolaris. Whereas in mangrove habitat are Avicennia alba, A. officinalis, Bruguiera gymnorrhiza, B. parvifolia, Rhizophora apiculata, and S. alba. They are recommended for habitat restoration.