Small update

Fixes