Issue #85 , fixed my test