Small update

Forgot small update