Fix generics.
Fix generics.

Kevin-Xi/learning_playground,Kevin-Xi/learning_playground,Kevin-Xi/learning_playground,Kevin-Xi/learning_playground,Kevin-Xi/learning_playground,Kevin-Xi/learning_playground