Unit test updated for address