Move private function to the right class