Add test for first assignment