TODO: Verify fix