Fixes issue with test

Should fix the jenkins testing timeout issue.