Fix lost code bug
dumb