Fixed code. Dang it.