Small fix to 5.1.0.

Think this is what's causing the issue but I'm not sure.