actually fix the small fix