Fix bad bug