Bug Fixes

Bla bla fixes