fix: forgot some changes