Attempted to fix the broken test