Bug fix: update the package name of the class to the correct one