Fix minor bug with install