small fix
fixed after some aggravation