small fix yo