fix code shorter