Fixes test error on Travis