Fix temp code that should have been removed.