Also fix the actual tests since it requires an array now