Fixing some code.