Fixed a part where I'd become confused