Cleaned up some unwanted code I missed