Fix running in travis