Fix minor mistake in the extends