Add test to check behavior