Fix simple example from the documentation (#105)

Fix simple example from the documentation (#105) #18