Fixed small bugs.