Small fix

Made a small fix