Fix bug
Document better