Fixes one more test