Minor bug fixes..