Solved bug (reference to old class name)