Updated for latest Nette