Quick bug fix

everything is so hardcoded lol...