solve little bug from last commit