fixing little bug