Simple fix to a potential problem

Simple fix to a potential problem