fixed the bug in the last commit

i'm dumb
working version