Fixed up test and everything looks good