Adding testcases covering the added check