Bug fix; Make test work, wasn't passing a parameter