Fixed somethign that was missed