Fixing another bug, oops