Fix #623, get enitity from current graph