Fixed issue with class visibility.