Completing unit test