1. Fix bug