Change package to match class in repo