Fixed Travis build, use test function