Add a test for the previous fix