Fixed this broken piece of shit