Fix bug/warning