Fixes bug that sneaked into prev fix