Put it back to how it was

Changed it a little bit.