Fixing the test, but it's not a long-term solution