Small fix
Fixes