Fixed up test and everything is passing =D