Added unit test for sanity check