Small fixes

forgot some stuff