Small fix/test