Fix/two tiny fixes