fixing issue #26

I'm not sure why you also removed this logic, but if `npm test` passes, then I guess it's fine.