Corrected position of code to keep vim folds fine