Bug fixes

@airbnb/di