Fixed minor bug (forgotten to update define...)