fix code a bit ty