fixing minor buggy