Updated the unit test to be something useful in testing