Fixed two small bugs.