Bug fixes

thar be bugs