fix typing errors from last commit