Sample unit test fixed