VIY: Bug fix

Bug fix