Fixed silly code error. (I wasn't able to test the code.)