Clean up code after fix.