Small fixes

Found two small problems, fixed.