Assert mistake in test class fixed