Fix compile error
Got demo working