fix minor error from last commit