Fix tests after add node changes