Fixed!

Sorry about that! Here's the fix. (cc @Jessassin )