change the test code a proper unit test