diff --git a/modelstore/__init__.py b/modelstore/__init__.py
index 87324397..399d457c 100644
--- a/modelstore/__init__.py
+++ b/modelstore/__init__.py
@@ -11,7 +11,7 @@
 #    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 importlib.metadata import version
+from importlib.metadata import PackageNotFoundError, version
 
 # pylint: disable=unused-import
 from modelstore.model_store import ModelStore
diff --git a/modelstore/models/model_manager.py b/modelstore/models/model_manager.py
index 443e6124..a4275ac5 100644
--- a/modelstore/models/model_manager.py
+++ b/modelstore/models/model_manager.py
@@ -225,7 +225,7 @@ def upload(
 
 def _format_numpy(model_params: dict) -> dict:
     for key, value in model_params.items():
-        if isinstance(value, (np.float16, np.float32, np.float64, np.float96)):
+        if isinstance(value, (np.float16, np.float32, np.float64)):
             model_params[key] = float(value)
         if isinstance(value, np.ndarray):
             model_params[key] = value.tolist()