Revert "Fix simple demo"

This reverts commit 9c0e3ca33b8ca903a63333c4065e51a579631541.