Fixed failing test on Travis-CI