Cleanup
Minor fix