Add back commented out code oopsie