fixed a lil bug