Fixed the correct class name