Simple bug fixes