Fix badly formatted code