Fix bug i created