quick fix

This is a quick fix complimentary to my last commit