fixed last minute issue before demo