fixed a small problem (i think)