Updates to fix travis