Update TreeAssistBlockListener.java