Bug fix java.lang.ClassCastException