Small demo fix