Fix bug in part II.