Fixing problem with story monad