Fixing one minor bug