Fix bug/remove comment