Changed a line to fit code style better