Fixed incorrect class name in framework integration