solve the doctest problem by changing top of the prism