Updated view handling
Fixed wrong class name