diff --git a/python/tamaas/__init__.py.in b/python/tamaas/__init__.py.in
index 3dfbe10..71d9d55 100644
--- a/python/tamaas/__init__.py.in
+++ b/python/tamaas/__init__.py.in
@@ -1,70 +1,70 @@
 # -*- mode:python; coding: utf-8 -*-
 # @file
 # @section LICENSE
 #
 # Copyright (©) 2016-2020 EPFL (École Polytechnique Fédérale de Lausanne),
 # Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as published
 # by the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
 # This program is distributed in the hope that it will be useful,
 # but WITHOUT ANY WARRANTY; without even the implied warranty of
 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 # GNU Affero General Public License for more details.
 #
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 """
 Tamaas is a library dedicated to the fast treatment of rough contact problems.
 
 See README.md and examples for guidance as how to use Tamaas.
 
 See __authors__, __license__, __copyright__ for extra information about Tamaas.
 """
 
 __version__ = "@version@"
 __author__ = @authors@
 # TODO Change copyright when is issue with unicode is found
 __copyright__ = u"Copyright (©) 2016-20 EPFL " \
     + u"(École Polytechnique Fédérale de Lausanne), " \
     + u"Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)"
 __license__ = "AGPLv3"
 __maintainer__ = "Lucas Frérot"
 __email__ = "@email@"
 
 
 try:
-    from . import _tamaas as _tm
+    from ._tamaas import model_type
     # Redefinition of model_type constants (for compatibility)
-    model_type_basic_1d = _tm.model_type.basic_1d
-    model_type_basic_2d = _tm.model_type.basic_2d
-    model_type_surface_1d = _tm.model_type.surface_1d
-    model_type_surface_2d = _tm.model_type.surface_2d
-    model_type_volume_1d = _tm.model_type.volume_1d
-    model_type_volume_2d = _tm.model_type.volume_2d
+    model_type_basic_1d = model_type.basic_1d
+    model_type_basic_2d = model_type.basic_2d
+    model_type_surface_1d = model_type.surface_1d
+    model_type_surface_2d = model_type.surface_2d
+    model_type_volume_1d = model_type.volume_1d
+    model_type_volume_2d = model_type.volume_2d
 
     from ._tamaas import *  # noqa
 
 except ImportError as e:
     print("Error trying to import _tamaas:\n{}".format(e))
     raise e
 
 
 class ParamHelper:
     """Legacy class to manage parameters/setters/getters"""
     def __init__(self, obj):
         self.obj = obj
 
     def set(self, params):
         for key in params:
             setter_name = 'set' + key
             try:
                 accessor = getattr(self.obj, setter_name)
                 accessor(params[key])
             except Exception:
                 print("Setter '{}({})' does not exist for object {}"
                       .format(setter_name, type(params[key]), self.obj))