Fixes bug introduced with last commit