Unit test updated to test