[BUG] minor design fix