Fix bug so you don't have to restart shell for it to work