fixs a little bug