Fixed based off of travis comments