Fix bug (#1)