Fix text code still WIP