Fixed a minor gotcha, which will cause some edge case to fail