just fixed a bit