diff --git a/model_compression_toolkit/core/pytorch/__init__.py b/model_compression_toolkit/core/pytorch/__init__.py
index a4d991ff0..424152e58 100644
--- a/model_compression_toolkit/core/pytorch/__init__.py
+++ b/model_compression_toolkit/core/pytorch/__init__.py
@@ -11,5 +11,4 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
-from model_compression_toolkit.core.pytorch.pytorch_device_config import set_working_device
\ No newline at end of file
+# ==============================================================================
\ No newline at end of file