Fix private class import