[TASK] Update Tests to work with Travis