- Fix minor bug