Hopefully fixed a bug.