Fix unittest for include_neighbors

Part of #416.