Fixed potential warning