Fix bug caused by previous commit