Add test to my previous fix