Fix formatting to restore functionality