Small bug fix.

Fixed small bugs I didn't initially find.