Fixed undefined variable :-/