Ooops, was testing state reloads and forgot to fix my code before commiting.

Ooops, was testing state reloads and forgot to fix my code before commiting.

This fixes the error I just pushed up.