fixed a little but