fix error from last commit