Quick fix for two small errors