fix bug after last commit