Fixed broken test
Forgot to update this in 1f67f7bbd91bf82ad166348dfec46b87c2a943f4