Add check for existance of root class