Fixes bug introduced in previous commit