Fixed mistake in code