ENH: Bug Fixes