Fixes bug 4