Fix thanks to travis testing