Adding tests for sanity check