Fix accidentally commented out code