protected function getNodeClass($type) { switch ($type) { case 'variable': return __NAMESPACE__ . '\\' . ucfirst($type) . 'NodeDefinition'; case 'bar': return __NAMESPACE__ . '\\' . ucfirst($type) . 'NodeDefinition'; default: return parent::getNodeClass($type); } }