Fix test error on Travis-CI, thanks to @greatman.