small fixes for demo