Minor

Bug fix