Fix bug.....