Fixes a broken test