Fixing one more test