Reversed some code that caused a error.