Fix bug when called as a command
v0.1.8