Little API fix
Little API fix
Fixed #49