Cleaned two lines of code