Bug Fix

Fixed the bug I mentioned in an earlier commit.