Fixed minor bug - note to self: Test code before commit.