Fixing something I needlessly broke :(