Updated for unit test

Updated tests to account for new and old URLS