Add test to show new behavior