I fixed a little bug