diff --git a/test/test_PhysicalProperty.py b/test/test_PhysicalProperty.py
index <HASH>..<HASH> 100644
--- a/test/test_PhysicalProperty.py
+++ b/test/test_PhysicalProperty.py
@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 from astropy import units
 import unittest
-from electrode import PhysicalProperty
+from physicalproperty import PhysicalProperty
 
 # Classes used in tests
 # =====================
@@ -56,12 +56,12 @@ class StandaloneInstantiation(unittest.TestCase):
 
     def test_set_Quantity_incompatible_units(self):
         """
-        Raise UnitsException when setting with Quantity of incompatible units.
+        Raise UnitsError when setting with Quantity of incompatible units.
         """
         pp = PhysicalProperty()
         qty = units.Quantity(100., "km")
         
-        self.assertRaises(units.UnitsException, pp.__set__, MockClassEmpty, qty)
+        self.assertRaises(units.UnitsError, pp.__set__, MockClassEmpty, qty)
 
     def test_set_numeric_above_up_bnd(self):
         """
Update test file for new API, new ver. of astropy