Small fixes

same thing here