Fixed the bug we just noticed