Fix for demo layout