<Minor> Re-added test class