Fixing some code shit.