Corrected class name in code example