Add test for variant