Tweaked test, fixed error