Fixed broken test, they are still bad though