Fixes

Fixed bug