Changed tree model listener type to fix the build