Add test for proper links