fixed mistake in class assignment