Fix bug introduced by bugfix