little fixes for demo