Fixed the exception coming up on the UI.